FIX: bug in raising informative in __main__

saqc/__main__.py

@@ -51,7 +51,7 @@ def readData(reader_dict, fname):
     reader = reader_dict.get(extension)
     if not reader:
         raise ValueError(
-            f"Unsupported file format '{extension}', use one of {tuple(reader.keys())}"
+            f"Unsupported file format '{extension}', use one of {tuple(reader_dict.keys())}"
         )
     return reader(fname)
 
@@ -61,7 +61,7 @@ def writeData(writer_dict, df, fname):
     writer = writer_dict.get(extension)
     if not writer:
         raise ValueError(
-            f"Unsupported file format '{extension}', use one of {tuple(writer.keys())}"
+            f"Unsupported file format '{extension}', use one of {tuple(writer_dict.keys())}"
         )
     writer(df, fname)
 

tests/test__main__.py

+#!/usr/bin/env python
+import pytest
+import numpy as np
+import pandas as pd
+
+
+def test_unknownFileExtention():
+    from saqc.__main__ import readData, writeData, setupIO
+
+    reader, writer = setupIO(np.nan)
+    with pytest.raises(ValueError):
+        readData(reader, "foo.unknown")
+    with pytest.raises(ValueError):
+        writeData(reader, pd.DataFrame(), "foo.unknown")