def test_with_generator(self): tmpX = (sample for sample in self.X) clf = GaussianNB() clf.fit(tmpX, self.Y) result = clf.predict([[-0.8, -1]]) self.assertEqual(result, [1])
def test_with_array(self): clf = GaussianNB() clf.fit(self.X, self.Y) result = clf.predict([[-0.8, -1]]) self.assertEqual(result, [1])