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