def test_argless_copyfuncs(self, func):
arr = cosmo_array(
np.ones((10, 10)),
units="Mpc",
cosmo_factor=cosmo_factor("a^1", 1),
comoving=False,
)
assert hasattr(getattr(arr, func)(), "cosmo_factor")
assert hasattr(getattr(arr, func)(), "comoving")
def test_unit_array(self):
arr = cosmo_array(
np.ones((10, 10)),
units="Mpc",
cosmo_factor=cosmo_factor("a^1", 1),
comoving=False,
)
res = arr.unit_array
assert hasattr(res, "cosmo_factor")
assert hasattr(res, "comoving")
def get_cosmo(dataset):
try:
cosmo_exponent = dataset.attrs["a-scale exponent"][0]
except:
# Can't load, 'graceful' fallback.
cosmo_exponent = 0.0
a_factor_this_dataset = a**cosmo_exponent
return cosmo_factor(a_factor_this_dataset, current_scale_factor)
def cosmo_factory(a_dependence):
return cosmo_factor(a_dependence, scale_factor)
