def test_get_classification_probabilities(self): training_set, sample = self.load_car_data() classifier = NaiveBayes(training_set) probabilities = classifier.get_classification_probabilities(sample) self.assertAlmostEqual(probabilities["yes"], 0.035, places=3) self.assertAlmostEqual(probabilities["no"], 0.070, places=3)