3.7

Author: kavun

Work/fileparse.py

@@ -5,13 +5,17 @@
 
 
 def parse_csv(
-    filename: str, select: list = [], types: list = [], has_headers: bool = True
+    filename: str,
+    select: list = [],
+    types: list = [],
+    has_headers: bool = True,
+    delimiter: str = ",",
 ) -> list:
     """
     Parse a CSV file into a list of records
     """
     with open(filename) as f:
-        rows = csv.reader(f)
+        rows = csv.reader(f, delimiter=delimiter)
         if has_headers:
             orig_headers = next(rows)
             headers = select if select else orig_headers

Work/fileparse_test.py

@@ -70,3 +70,15 @@ def test_parse_csv_no_headers_3_6():
         ("WMT", 49.74),
         ("XOM", 69.35),
     ]
+
+
+def test_parse_csv_delimiter_3_7():
+    assert parse_csv("Data/portfolio.dat", types=[str, int, float], delimiter=" ") == [
+        { "name": "AA", "shares": 100, "price": 32.2 },
+        { "name": "IBM", "shares": 50, "price": 91.1 },
+        { "name": "CAT", "shares": 150, "price": 83.44 },
+        { "name": "MSFT", "shares": 200, "price": 51.23 },
+        { "name": "GE", "shares": 95, "price": 40.37 },
+        { "name": "MSFT", "shares": 50, "price": 65.1 },
+        { "name": "IBM", "shares": 100, "price": 70.44 },
+    ]