def test_slices(self):
ds = DataSet(range(0, 120))
self.assertEqual(range(0, 30), ds.slice(0).raw_data())
self.assertEqual(range(30, 60), ds.slice(1).raw_data())
self.assertEqual(range(60, 90), ds.slice(2).raw_data())
self.assertEqual(range(90, 120), ds.slice(3).raw_data())
ds = DataSet(range(0, 120), slice_size=1)
self.assertEqual([0], ds.slice(0).raw_data())
self.assertEqual([119], ds.slice(119).raw_data())
def test_slices(self):
def mat_repeated(repeated):
basic = "1 2;3 4;5 6;7 8;9 10;"
return numpy.matrix("[%s]" % (basic * repeated).strip(";"))
ds = DataSet(mat_repeated(10))
self.assertTrue((mat_repeated(1) == ds.slice(0).raw_data()).all())
self.assertTrue((mat_repeated(1) == ds.slice(1).raw_data()).all())
self.assertTrue((mat_repeated(1) == ds.slice(2).raw_data()).all())
ds = DataSet(mat_repeated(10), slice_size=10)
self.assertTrue((mat_repeated(2) == ds.slice(0).raw_data()).all())
ds = DataSet(mat_repeated(10), slice_size=100)
self.assertTrue((mat_repeated(10) == ds.slice(0).raw_data()).all())
