Extraneous columnns expected in csv dictionary in test function are r…

…emoved
lanl · Nov 15, 2024 · 7fe1964 · 7fe1964
1 parent d9cd059
commit 7fe1964
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dsi/plugins/tests/test_file_reader.py b/dsi/plugins/tests/test_file_reader.py
@@ -56,8 +56,8 @@ def test_csv_plugin_adds_rows():
     for key, val in plug.output_collector["Csv"].items():
         assert len(val) == 4
 
-    # 11 Csv cols + 1 inherited FileReader cols
-    assert len(plug.output_collector["Csv"].keys()) == 12
+    # 11 Csv cols
+    assert len(plug.output_collector["Csv"].keys()) == 11
 
 def test_csv_plugin_adds_rows_multiple_files():
     path1 = '/'.join([get_git_root('.'), 'examples/data', 'wildfiredata.csv'])
@@ -69,8 +69,8 @@ def test_csv_plugin_adds_rows_multiple_files():
     for key, val in plug.output_collector["Csv"].items():
         assert len(val) == 8
 
-    # 13 Csv cols + 2 inherited FileReader cols
-    assert len(plug.output_collector["Csv"].keys()) == 15
+    # 13 Csv cols
+    assert len(plug.output_collector["Csv"].keys()) == 13
 
 def test_csv_plugin_adds_rows_multiple_files_strict_mode():
     path1 = '/'.join([get_git_root('.'), 'examples/data', 'wildfiredata.csv'])