def test_hash(self):
# for slice_size in [30, 399, 1, 5119]:
for slice_size in [30, 1]:
ds1 = MatFileDataSet(self.__FILE, slice_size=slice_size)
ds2 = MatFileDataSet(self.__FILE, slice_size=slice_size)
self.assertEqual(ds1.hash(), ds2.hash())
for i in range(0, len(ds1), 100):
self.assertEqual(ds1.hash(i), ds2.hash(i))
slice1 = ds1.slice_from_hash(ds1.hash(i)).raw_data()
slice2 = ds2.slice_from_hash(ds2.hash(i)).raw_data()
slice12 = ds1.slice_from_hash(ds2.hash(i)).raw_data()
slice21 = ds2.slice_from_hash(ds1.hash(i)).raw_data()
self.assertTrue((slice1 == slice2).all())
self.assertTrue((slice12 == slice21).all())
self.assertTrue((slice1 == slice21).all())
def test_slices(self):
ds1 = MatFileDataSet(self.__FILE)
ds2 = MatFileDataSet(self.__FILE, slice_size=20)
ds3 = MatFileDataSet(self.__FILE, slice_size=50)
ds4 = MatFileDataSet(self.__FILE, slice_size=99)
ds5 = MatFileDataSet(self.__FILE, slice_size=100)
self.assertTrue((MAT1 == ds1.slice(0).raw_data()[0:6][:, 0:3]).all())
self.assertTrue((MAT1 == ds2.slice(0).raw_data()[0:6][:, 0:3]).all())
self.assertTrue((MAT1 == ds3.slice(0).raw_data()[0:6][:, 0:3]).all())
self.assertTrue((MAT2 == ds2.slice(100).raw_data()[0:6][:, 3:6]).all())
self.assertTrue((MAT2 == ds3.slice(40).raw_data()[0:6][:, 3:6]).all())
self.assertTrue((MAT2 == ds5.slice(20).raw_data()[0:6][:, 3:6]).all())
self.assertTrue((MAT3 == ds4.slice(15).raw_data()[0:6][:, 6:9]).all())
