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"))