def test_dataset_iterator(self): """Dataset must correctly work as iterable object.""" dts = Dataset() N = 10 for i in range(N): dts.add_row([i]) for i in range(2): self.assertEqual(N, len([i for i in dts]))
def test_get_dataset_columns(self): """Test column splitting correctness. | 0 | 1 | 2 | | 3 | 4 | 5 | """ dts = Dataset() [dts.add_row(i) for i in chunk(range(6), 3)] for i in range(3): self.assertEqual([i, i + 3], list(dts.column(i))) # test multiple self.assertEqual([(0, 1), (3, 4)], list(dts.column(0, 1))) self.assertEqual([(1, 2), (4, 5)], list(dts.column(1, 2)))
def test_get_dataset_columns(self): """Test column splitting correctness. | 0 | 1 | 2 | | 3 | 4 | 5 | """ dts = Dataset() [dts.add_row(i) for i in chunk(range(6), 3)] for i in range(3): self.assertEqual([i, i + 3], list(dts.column(i))) # test multiple self.assertEqual([(0, 1), (3, 4)], list(dts.column(0,1))) self.assertEqual([(1, 2), (4, 5)], list(dts.column(1,2)))