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