def test_get_label_empty_set(self):
dataset = DataSet([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]])
assert_that(dataset.get_label_set(), has_length(0))
def test_get_label_set(self):
dataset = DataSet([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]],
labels=["a", "b", "b", "c", "a", "b"])
assert_that(dataset.get_label_set(), contains_inanyorder("a", "b", "c"))
