def test_fit(self): clf = CentroidClassifier() clf.fit(self.X, self.Y) np.testing.assert_almost_equal(clf.similarity_index.index[1], [-2, -1.33333], decimal=5)
def test_predict(self): clf = CentroidClassifier() clf.fit(self.X, self.Y) result = clf.predict([[-0.8, -1]]) self.assertEqual(result, [2])