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