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')