Example #1
0
 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"))
Example #2
0
 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))