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])