def test_sorted_and_unsorted(self): with h5py.File(test_h5_file_path, mode='r') as h5_f: usi_dset = USIDataset(h5_f['/Raw_Measurement/source_main']) nd_slow_to_fast, nd_fast_to_slow = self.get_expected_n_dim(h5_f) actual_f2s = usi_dset.get_n_dim_form(lazy=False) self.assertTrue(np.allclose(nd_fast_to_slow, actual_f2s)) nd_form, success = reshape_to_n_dims(usi_dset, sort_dims=True) print(nd_form.shape) usi_dset.toggle_sorting() actual_s2f = usi_dset.get_n_dim_form(lazy=False) self.assertTrue(np.allclose(nd_slow_to_fast, actual_s2f))
def test_unsorted(self): with h5py.File(test_h5_file_path, mode='r') as h5_f: h5_main = h5_f['/Raw_Measurement/source_main'] expected = np.reshape(h5_main, (3, 5, 7, 2)) expected = np.transpose(expected, (1, 0, 2, 3)) usi_dset = USIDataset(h5_main) self.assertTrue(np.allclose(expected, usi_dset.get_n_dim_form(lazy=False)))
def test_get_n_dim_form_sorted(self): with h5py.File(test_h5_file_path, mode='r') as h5_f: h5_main = h5_f['/Raw_Measurement/source_main'] expected = np.reshape(h5_main, (3, 5, 7, 2)) expected = np.transpose(expected, (1, 0, 3, 2)) pycro_dset = USIDataset(h5_main) pycro_dset.toggle_sorting() self.assertTrue(np.allclose(expected, pycro_dset.get_n_dim_form()))