def test_classify_all(self):
training_set = DataSet([[1, 1], [2, 2], [11, 11], [12, 12]],
labels=["a", "a", "b", "b"])
classifier = Knn(training_set, k=3)
dataset = [[1.5, 1.3], [12.2, 12.9]]
classes = classifier.classify_all(dataset).get_classifications()
assert_that(classes, contains("a", "b"))
