def _test_pmf(self, name, result_shape): target_path = os.path.join(self.tempdir, '%s.hdf5' % name) disagg_subsets.extract_subsets( 111, self.SITE, self.full_matrix_path, self.LATITUDE_BIN_LIMITS, self.LONGITUDE_BIN_LIMITS, self.MAGNITUDE_BIN_LIMITS, self.EPSILON_BIN_LIMITS, self.DISTANCE_BIN_LIMITS, target_path, [name] ) expected_result = self.read_data_file('%s.dat' % name, result_shape) result = h5py.File(target_path, 'r')[name].value helpers.assertDeepAlmostEqual(self, expected_result, result)
def test_multiple_matrices(self): target_path = os.path.join(self.tempdir, 'multiple.hdf5') pmfs = { 'MagDistEpsPMF': ('MagDistEpsPMF.dat', [self.NMAG - 1, self.NDIST - 1, self.NEPS - 1]), 'LatLonPMF': ('LatLonPMF.dat', [self.NLAT - 1, self.NLON - 1]), disagg.FULL_DISAGG_MATRIX: (self.FULL_MATRIX_DATA, self.FULL_MATRIX_SHAPE) } disagg_subsets.extract_subsets( 112, self.SITE, self.full_matrix_path, self.LATITUDE_BIN_LIMITS, self.LONGITUDE_BIN_LIMITS, self.MAGNITUDE_BIN_LIMITS, self.EPSILON_BIN_LIMITS, self.DISTANCE_BIN_LIMITS, target_path, pmfs.keys() ) result = h5py.File(target_path, 'r') for name, (datafile, shape) in pmfs.items(): expected_result = self.read_data_file(datafile, shape) helpers.assertDeepAlmostEqual(self, expected_result, result[name])