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