def test_calc_prob_feature_given_class(self): training_set, sample = self.load_car_data() classifier = NaiveBayes(training_set) prob = classifier._calc_prob_feature_given_class("yes", "color", sample["color"]) self.assertAlmostEqual(prob, 0.57, places=2)