Commit 0b0cd797 authored by Hongbin Yang's avatar Hongbin Yang
Browse files

Add one-sample test for metric

parent 9ee46f4e
Loading
Loading
Loading
Loading
+15 −0
Original line number Diff line number Diff line
@@ -23,6 +23,21 @@ class MetricsTest(googletest.TestCase):
                                    1.0 - expected_agreement)
    self.assertAlmostEqual(kappa, expected_kappa)

  def test_one_sample(self):
    np.random.seed(123)
    n_samples = 2
    y_true = np.array([0, 0])
    y_pred = np.random.rand(n_samples, 2)
    w = np.array([0, 1])
    all_metrics = [
      dc.metrics.Metric(dc.metrics.recall_score),
      dc.metrics.Metric(dc.metrics.matthews_corrcoef),
      dc.metrics.Metric(dc.metrics.roc_auc_score)
    ]
    for metric in all_metrics:
      score = metric.compute_singletask_metric(y_true, y_pred, w)
      self.assertTrue(np.isnan(score) or score==0)

  def test_r2_score(self):
    """Test that R^2 metric passes basic sanity tests"""
    np.random.seed(123)