def test_reduce_grid_info():
seds_trim_fname = download_rename("beast_example_phat_seds_trim.grid.hd5")
sub_fnames = subgridding_tools.split_grid(seds_trim_fname, 3)
complete_g_info = subgridding_tools.subgrid_info(seds_trim_fname)
cap_unique = 50
sub_g_info = subgridding_tools.reduce_grid_info(sub_fnames,
nprocs=3,
cap_unique=cap_unique)
for q in complete_g_info:
if q not in sub_g_info:
raise AssertionError()
if not complete_g_info[q]["min"] == sub_g_info[q]["min"]:
raise AssertionError()
if not complete_g_info[q]["max"] == sub_g_info[q]["max"]:
raise AssertionError()
num_unique = len(complete_g_info[q]["unique"])
if num_unique > cap_unique:
# Cpan still be larger if one of the sub results during the
# reduction is larger. This is as intended.
if not sub_g_info[q]["num_unique"] >= cap_unique:
raise AssertionError()
else:
if not sub_g_info[q]["num_unique"] == num_unique:
raise AssertionError()
def test_reduce_grid_info(self):
"""
Split a cached version of a sed grid and check that [not quite
sure what this is checking - details needed].
"""
sub_fnames = subgridding_tools.split_grid(self.seds_trim_fname_cache,
3)
complete_g_info = subgridding_tools.subgrid_info(
self.seds_trim_fname_cache)
cap_unique = 50
sub_g_info = subgridding_tools.reduce_grid_info(sub_fnames,
nprocs=3,
cap_unique=cap_unique)
for q in complete_g_info:
if q not in sub_g_info:
raise AssertionError()
if not complete_g_info[q]["min"] == sub_g_info[q]["min"]:
raise AssertionError()
if not complete_g_info[q]["max"] == sub_g_info[q]["max"]:
raise AssertionError()
num_unique = len(complete_g_info[q]["unique"])
if num_unique > cap_unique:
# Cpan still be larger if one of the sub results during the
# reduction is larger. This is as intended.
if not sub_g_info[q]["num_unique"] >= cap_unique:
raise AssertionError()
else:
if not sub_g_info[q]["num_unique"] == num_unique:
raise AssertionError()
def test_reduce_grid_info():
seds_trim_fname = download_rename('beast_example_phat_seds_trim.grid.hd5')
sub_fnames = subgridding_tools.split_grid(seds_trim_fname, 3)
complete_g_info = subgridding_tools.subgrid_info(seds_trim_fname)
cap_unique = 50
sub_g_info = subgridding_tools.reduce_grid_info(
sub_fnames, nprocs=3, cap_unique=cap_unique)
for q in complete_g_info:
assert q in sub_g_info
assert complete_g_info[q]['min'] == sub_g_info[q]['min']
assert complete_g_info[q]['max'] == sub_g_info[q]['max']
num_unique = len(complete_g_info[q]['unique'])
if num_unique > cap_unique:
# Cpan still be larger if one of the sub results during the
# reduction is larger. This is as intended.
assert sub_g_info[q]['num_unique'] >= cap_unique
else:
assert sub_g_info[q]['num_unique'] == num_unique