def test_ib_override_z_heat_max(perturb_field):
# save previous z_heat_max
zheatmax = wrapper.global_params.Z_HEAT_MAX
wrapper.ionize_box(
redshift=perturb_field.redshift,
perturbed_field=perturb_field,
z_heat_max=12.0,
)
assert wrapper.global_params.Z_HEAT_MAX == zheatmax
def test_ib_from_z(default_user_params, perturb_field):
ib = wrapper.ionize_box(
redshift=perturb_field.redshift,
user_params=default_user_params,
regenerate=False,
)
assert ib.redshift == perturb_field.redshift
assert ib.user_params == perturb_field.user_params
assert ib.cosmo_params == perturb_field.cosmo_params
assert ib.cosmo_params is not perturb_field.cosmo_params
def test_ib_new_seed(ionize_box_lowz, perturb_field_lowz, tmpdirec):
# this should fail because perturb_field has a seed set already, which isn't 1.
with pytest.raises(ValueError):
wrapper.ionize_box(
perturbed_field=perturb_field_lowz,
random_seed=1,
)
ib = wrapper.ionize_box(
cosmo_params=perturb_field_lowz.cosmo_params,
redshift=perturb_field_lowz.redshift,
user_params=perturb_field_lowz.user_params,
random_seed=1,
)
# we didn't write it, and this has a different seed
assert not ib.exists(direc=tmpdirec)
assert ib.random_seed != ionize_box_lowz.random_seed
assert not np.all(ib.xH_box == ionize_box_lowz.xH_box)
def test_ib_no_z(ic):
with pytest.raises(ValueError):
wrapper.ionize_box(init_boxes=ic)
def ionize_box_lowz(perturb_field_lowz):
"""A default ionize_box at lower redshift."""
return wrapper.ionize_box(perturbed_field=perturb_field_lowz, write=True)
def test_ib_bad_st(ic, redshift):
with pytest.raises(ValueError):
wrapper.ionize_box(redshift=redshift, spin_temp=ic)
def ionize_box(perturb_field):
"""A default ionize_box"""
return wrapper.ionize_box(perturbed_field=perturb_field, write=True)
def test_ib_override_z(perturb_field):
with pytest.raises(ValueError):
wrapper.ionize_box(
redshift=perturb_field.redshift + 1,
perturbed_field=perturb_field,
)
def test_ib_from_pf(perturb_field):
ib = wrapper.ionize_box(perturbed_field=perturb_field)
assert ib.redshift == perturb_field.redshift
assert ib.user_params == perturb_field.user_params
assert ib.cosmo_params == perturb_field.cosmo_params
