Add tests for multiple columns and negative floats

Author: kleinjohann

neo/test/rawiotest/test_phyrawio.py

@@ -31,26 +31,35 @@ def test_csv_tsv_parser_with_csv(self):
         csv_tempfile = Path(tempfile.gettempdir()).joinpath('test.csv')
         with open(csv_tempfile, 'w') as csv_file:
             csv_writer = csv.writer(csv_file, delimiter=',')
-            csv_writer.writerow(['cluster_id', 'some_annotation'])
-            csv_writer.writerow([1, 'Good'])
-            csv_writer.writerow([2, 10])
-            csv_writer.writerow([3, 1.23])
+            csv_writer.writerow(['cluster_id', 'some_annotation', 'some_other_annotation'])
+            csv_writer.writerow([1, 'Good', 'Bad'])
+            csv_writer.writerow([2, 10, -2])
+            csv_writer.writerow([3, 1.23, -0.38])
 
         # the parser in PhyRawIO runs csv.DictReader to parse the file
         # csv.DictReader for python version 3.6+ returns list of OrderedDict
         if (3, 6) <= sys.version_info < (3, 8):
             target = [OrderedDict({'cluster_id': 1,
-                                   'some_annotation': 'Good'}),
+                                   'some_annotation': 'Good',
+                                   'some_other_annotation': 'Bad'}),
                       OrderedDict({'cluster_id': 2,
-                                   'some_annotation': 10}),
+                                   'some_annotation': 10,
+                                   'some_other_annotation': -2}),
                       OrderedDict({'cluster_id': 3,
-                                   'some_annotation': 1.23})]
+                                   'some_annotation': 1.23,
+                                   'some_other_annotation': -0.38})]
 
         # csv.DictReader for python version 3.8+ returns list of dict
         elif sys.version_info >= (3, 8):
-            target = [{'cluster_id': 1, 'some_annotation': 'Good'},
-                      {'cluster_id': 2, 'some_annotation': 10},
-                      {'cluster_id': 3, 'some_annotation': 1.23}]
+            target = [{'cluster_id': 1,
+                       'some_annotation': 'Good',
+                       'some_other_annotation': 'Bad'},
+                      {'cluster_id': 2,
+                       'some_annotation': 10,
+                       'some_other_annotation': -2},
+                      {'cluster_id': 3,
+                       'some_annotation': 1.23,
+                       'some_other_annotation': -0.38}]
 
         list_of_dict = PhyRawIO._parse_tsv_or_csv_to_list_of_dict(csv_tempfile)