def test_get_sample_ids_named(self): dataset = DataSet(pd.DataFrame([[1, 2], [3, 4], [5, 6]], index=["a", "b", "c"])) assert_that(dataset.get_sample_ids(), contains("a", "b", "c"))
def test_get_sample_ids(self): dataset = DataSet([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) assert_that(dataset.get_sample_ids(), contains(0, 1, 2))