def test_calc_prob_class(self):
training_set = DataSet([[1, 2], [3, 4], [5, 6], [7, 8]],
labels=["cat", "dog", "cat", "cat"])
classifier = NaiveBayes(training_set)
prob = classifier._calc_prob_class("cat")
self.assertAlmostEqual(prob, 0.75, places=2)
