Update tests

Author: thieu1995

tests/test_classification.py

@@ -10,7 +10,7 @@
 np.random.seed(42)
 
 
-def test_RegressionMetric_class():
+def test_ClassificationMetric_class():
     y_true = [0, 1, 0, 0, 1, 0]
     y_pred = [0, 1, 0, 0, 0, 1]
 

tests/test_clustering.py

@@ -10,10 +10,9 @@
 np.random.seed(42)
 
 
-def test_RegressionMetric_class():
+def test_internal_ClusteringMetric_class():
     # generate sample data
     X = np.random.uniform(-1, 10, size=(300, 6))
-    y_true = np.random.randint(0, 3, size=300)
     y_pred = np.random.randint(0, 3, size=300)
 
     evaluator = ClusteringMetric(y_pred=y_pred, X=X, decimal=5)
@@ -23,3 +22,17 @@ def test_RegressionMetric_class():
 
     assert bhi1 == bhi2
     assert bhi2 == bhi3
+
+
+def test_external_ClusteringMetric_class():
+    # generate sample data
+    y_true = np.random.randint(0, 3, size=300)
+    y_pred = np.random.randint(0, 3, size=300)
+
+    evaluator = ClusteringMetric(y_true=y_true, y_pred=y_pred, decimal=5)
+    prs1 = evaluator.precision_score()
+    prs2 = evaluator.PrS(y_true=y_true, y_pred=y_pred)
+    prs3 = evaluator.PrS()
+
+    assert prs1 == prs2
+    assert prs2 == prs3