Extended testing

LucaCappelletti94 · Dec 4, 2019 · 438968f · 438968f
1 parent e04ac2a
commit 438968f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/test_barplots/val_auroc.jpg b/test_barplots/val_auroc.jpg
diff --git a/tests/test_histograms.py b/tests/test_histograms.py
@@ -99,3 +99,13 @@ def test_histograms():
         path += "_{feature}.jpg"
 
         barplots(**kwargs, path=path, custom_defaults=custom_defaults)
+
+
+def test_single_index():
+    root = "test_barplots"
+    df = pd.read_csv("tests/test_case.csv")
+    barplots(
+        df,
+        ["cell_line"],
+        path="{root}/{{feature}}.jpg".format(root=root)
+    )
diff --git a/tests/test_wrong_parameters.py b/tests/test_wrong_parameters.py
@@ -4,7 +4,7 @@
 
 def test_wrong_parameters():
     root = "test_barplots"
-    df = pd.read_csv("tests/test_case.csv", index_col=0)
+    df = pd.read_csv("tests/test_case.csv")
     with pytest.raises(ValueError):
         barplots(
             df,