def test_at_100(self): binary_scores = binarize_scores_at(self.scores, 1.0) expected = np.array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1]) np.testing.assert_equal(binary_scores, expected)
def test_at_50(self): binary_scores = binarize_scores_at(self.scores, 0.5) expected = np.array([1, 1, 1, 1, 1, 0, 0, 0, 0, 0]) np.testing.assert_equal(binary_scores, expected)