class TestNaiveBayes(unittest.TestCase):
""" Testing Naive Bayes """
def setUp(self):
self.nb = NaiveBayes('tennis.csv')
self.nb.fit()
dict_input = {'Sky': 'Sunny', 'Temp': 'Hot', 'Humid': 'High', 'Wind': 'Weak'}
self.input_ = pd.Series(dict_input)
def test_single_row_input_prob(self):
self.assertAlmostEqual(self.nb.predict(self.input_, probability=True), 0.241975, places=6) # hand calculated probability
def test_single_row_input_answer(self):
self.assertEqual(self.nb.predict(self.input_), "No") # hand calculated probability
def setUp(self):
self.nb = NaiveBayes('tennis.csv')
self.nb.fit()
dict_input = {'Sky': 'Sunny', 'Temp': 'Hot', 'Humid': 'High', 'Wind': 'Weak'}
self.input_ = pd.Series(dict_input)
