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)