test: added incorrect object passed to saveDataframeCSV and fixed out…

…putFilePath error test function call
bhklab · Dec 18, 2023 · 281637b · 281637b
1 parent 69a0d84
commit 281637b
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/test_metadata.py b/tests/test_metadata.py
@@ -56,6 +56,18 @@ def test_saveDataframeCSV_outputFilePath_error(nsclcSummaryFilePath):
                                    segType = "SEG")
     badFilePath = "notacsv.xlsx"
     with pytest.raises(ValueError):
-        matchCTtoSegmentation(nsclcSummaryFilePath, segType = 'SEG',
-                              outputFilePath = badFilePath)
+        saveDataframeCSV(testDataframe, badFilePath)
 
+@pytest.mark.parametrize(
+    "notADataFrame",
+    [
+        ['list', 'of', 'features'],
+        "Just a string",
+        {"feat1": 34, "feat2": 10000, "feat3": 3.141592}
+    ]
+)
+def test_saveDataframeCSV_dataframe_error(notADataFrame):
+    """Check ValueError is raised when something other than pd.DataFrame is passed"""
+    goodFilePath = "tests/output/badDataframeExample.csv"
+    with pytest.raises(ValueError):
+        saveDataframeCSV(notADataFrame, goodFilePath)