feature: test single numpy array

alistairewj · Dec 15, 2019 · bbc0a65 · bbc0a65
1 parent dd56ce9
commit bbc0a65
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_pyroc.py b/tests/test_pyroc.py
@@ -36,6 +36,16 @@ def test_pyroc_input_parsing(ovarian_cancer_dataset):
     target = df['outcome'].values
     df = df.drop('outcome', axis=1)
 
+    # single numpy array of preds
+    preds = df['albumin'].values
+    roc = pyroc.ROC(
+        target,
+        preds,
+    )
+    # since roc wasn't provided predictor labels,
+    # keys are integers (i)
+    assert (roc.preds[0][:n] == expected_values['albumin'][:n]).all()
+
     # numpy arrays
     preds = df.values
     roc = pyroc.ROC(