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