Jared2

Author: exAdun

test/test_roc.py

@@ -0,0 +1,13 @@
+import numpy as np
+import pytest
+from toolkit.plot import plot_roc_curve
+
+@pytest.fixture()
+def binary_classification_data():
+    y_true = np.random.randint(low=0, high=2, size=100)
+    y_pred = np.random.rand(100)
+    return y_true, y_pred
+
+def test_plot_roc_curve(binary_classification_data):
+    y_true, y_pred = binary_classification_data
+    plot_roc_curve(y_true, y_pred)

toolkit/plot.py

@@ -169,7 +169,7 @@ def wordcloudviz(column):
     plt.tight_layout(pad=0)
     plt.show()
 >>>>>>> 0f8d8abe7a26aac02e768b21b22a1d2e58bd6d30
-
+ 
 def plot_cumulative_variance_ratio(pca, n_features):
 
     '''