def test_custom_particle_def():
ytcfg.add_section('ramses-particles')
ytcfg['ramses-particles', 'fields'] = '''particle_position_x, d
particle_position_y, d
particle_position_z, d
particle_velocity_x, d
particle_velocity_y, d
particle_velocity_z, d
particle_mass, d
particle_identifier, i
particle_refinement_level, I
particle_birth_time, d
particle_foobar, d
'''
ds = yt.load(ramsesCosmo)
def check_unit(array, unit):
assert str(array.in_cgs().units) == unit
try:
assert ('io', 'particle_birth_time') in ds.derived_field_list
assert ('io', 'particle_foobar') in ds.derived_field_list
check_unit(ds.r['io', 'particle_birth_time'], 's')
check_unit(ds.r['io', 'particle_foobar'], 'dimensionless')
finally:
ytcfg.remove_section('ramses-particles')
def test_custom_hydro_def():
ytcfg.add_section("ramses-hydro")
ytcfg[
"ramses-hydro", "fields"
] = """
Density
x-velocity
y-velocity
z-velocity
Foo
Bar
"""
ds = yt.load(ramsesCosmo)
def check_unit(array, unit):
assert str(array.in_cgs().units) == unit
try:
assert ("ramses", "Foo") in ds.derived_field_list
assert ("ramses", "Bar") in ds.derived_field_list
check_unit(ds.r["ramses", "Foo"], "dimensionless")
check_unit(ds.r["ramses", "Bar"], "dimensionless")
finally:
ytcfg.remove_section("ramses-hydro")
def test_custom_particle_def():
ytcfg.add_section("ramses-particles")
ytcfg["ramses-particles", "fields"] = """particle_position_x, d
particle_position_y, d
particle_position_z, d
particle_velocity_x, d
particle_velocity_y, d
particle_velocity_z, d
particle_mass, d
particle_identifier, i
particle_refinement_level, I
particle_birth_time, d
particle_foobar, d
"""
ds = yt.load(ramsesCosmo)
def check_unit(array, unit):
assert str(array.in_cgs().units) == unit
try:
assert ("io", "particle_birth_time") in ds.derived_field_list
assert ("io", "particle_foobar") in ds.derived_field_list
check_unit(ds.r["io", "particle_birth_time"], "s")
check_unit(ds.r["io", "particle_foobar"], "dimensionless")
finally:
ytcfg.remove_section("ramses-particles")
def custom_ramses_fields_conf():
old_config = deepcopy(ytcfg.config_root.as_dict())
ytcfg.add_section("ramses-hydro")
ytcfg.add_section("ramses-grav")
ytcfg.set("ramses-hydro", "fields", custom_hydro)
ytcfg.set("ramses-grav", "fields", custom_grav)
yield
ytcfg.remove_section("ramses-hydro")
ytcfg.remove_section("ramses-grav")
ytcfg.update(old_config)
def test_custom_hydro_def():
ytcfg.add_section('ramses-hydro')
ytcfg['ramses-hydro', 'fields'] = '''
Density
x-velocity
y-velocity
z-velocity
Foo
Bar
'''
ds = yt.load(ramsesCosmo)
def check_unit(array, unit):
assert str(array.in_cgs().units) == unit
try:
assert ('ramses', 'Foo') in ds.derived_field_list
assert ('ramses', 'Bar') in ds.derived_field_list
check_unit(ds.r['ramses', 'Foo'], 'dimensionless')
check_unit(ds.r['ramses', 'Bar'], 'dimensionless')
finally:
ytcfg.remove_section('ramses-hydro')
