def test_AHFHalosDataset():
ds = load(ahf_halos)
assert isinstance(ds, AHFHalosDataset)
ad = ds.all_data()
ad[("all", "particle_mass")]
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_snapshot_033():
ds = data_dir_load(os33)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
for test in sph_answer(ds, "snap_033", 2 * 128**3, _fields):
test_snapshot_033.__name__ = test.description
yield test
def test_particle_subselection():
#This checks that we correctly subselect from a dataset, first by making
#sure we get all the particles, then by comparing manual selections against
#them.
ds = data_dir_load(snap_33)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_gasoline_dmonly():
cosmology_parameters = dict(current_redshift=0.0,
omega_lambda=0.728,
omega_matter=0.272,
hubble_constant=0.702)
kwargs = dict(cosmology_parameters=cosmology_parameters,
unit_base={'length': (60.0, "Mpccm/h")})
ds = data_dir_load(gasoline_dmonly, TipsyDataset, (), kwargs)
for test in nbody_answer(ds, "agora_1e11.00400", 10550576, _fields):
yield test
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_particle_selection():
ds = data_dir_load(dmonly)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_arepo_tng59_selection():
ds = data_dir_load(tng59_h5)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_arepo_hdf5_selection():
ds = data_dir_load(bullet_h5)
assert isinstance(ds, ArepoHDF5Dataset)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_EagleDataset():
ds = data_dir_load(s28)
assert isinstance(ds, EagleDataset)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_arepo_tng59_selection():
ds = data_dir_load(tng59_h5, kwargs={"bounding_box": _tng59_bbox})
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_arepo_hdf5_selection():
ds = data_dir_load(bullet_h5)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_FLASHParticleDataset_selection():
ds = data_dir_load(fid_1to3_b1)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_non_cosmo_dataset_selection():
ds = load(keplerian_ring)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_cosmo_dataset_selection():
ds = load(EAGLE_6)
psc = ParticleSelectionComparison(ds)
psc.run_defaults()
def test_pkdgrav():
ds = data_dir_load(pkdgrav, TipsyDataset, (), kwargs=pkdgrav_kwargs)
for test in nbody_answer(ds, "halo1e11_run1.00400", 26847360, _fields):
yield test
psc = ParticleSelectionComparison(ds)
psc.run_defaults()