def test_old_spatial_data():
ds = data_dir_load(enzotiny)
fn = "DD0046_proj.h5"
full_fn = os.path.join(ytdata_dir, fn)
proj_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, proj_ds)
assert isinstance(proj_ds, YTSpatialPlotDataset)
yield YTDataFieldTest(full_fn, ("grid", "density"), geometric=False)
def test_old_profile_data():
tmpdir = tempfile.mkdtemp()
curdir = os.getcwd()
os.chdir(tmpdir)
ds = data_dir_load(enzotiny)
ad = ds.all_data()
profile_1d = create_profile(
ad,
("gas", "density"),
("gas", "temperature"),
weight_field=("gas", "cell_mass"),
)
fn = "DD0046_Profile1D.h5"
full_fn = os.path.join(ytdata_dir, fn)
prof_1d_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, prof_1d_ds)
assert isinstance(prof_1d_ds, YTProfileDataset)
for field in profile_1d.standard_deviation:
assert_array_equal(
profile_1d.standard_deviation[field],
prof_1d_ds.profile.standard_deviation["data", field[1]],
)
p1 = ProfilePlot(
prof_1d_ds.data,
("gas", "density"),
("gas", "temperature"),
weight_field=("gas", "cell_mass"),
)
p1.save()
yield YTDataFieldTest(full_fn, ("gas", "temperature"), geometric=False)
yield YTDataFieldTest(full_fn, ("index", "x"), geometric=False)
yield YTDataFieldTest(full_fn, ("gas", "density"), geometric=False)
fn = "DD0046_Profile2D.h5"
full_fn = os.path.join(ytdata_dir, fn)
prof_2d_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, prof_2d_ds)
assert isinstance(prof_2d_ds, YTProfileDataset)
p2 = PhasePlot(
prof_2d_ds.data,
("gas", "density"),
("gas", "temperature"),
("gas", "cell_mass"),
weight_field=None,
)
p2.save()
yield YTDataFieldTest(full_fn, ("gas", "density"), geometric=False)
yield YTDataFieldTest(full_fn, ("index", "x"), geometric=False)
yield YTDataFieldTest(full_fn, ("gas", "temperature"), geometric=False)
yield YTDataFieldTest(full_fn, ("index", "y"), geometric=False)
yield YTDataFieldTest(full_fn, ("gas", "cell_mass"), geometric=False)
os.chdir(curdir)
shutil.rmtree(tmpdir)
def test_old_datacontainer_data():
ds = data_dir_load(enzotiny)
sphere = ds.sphere(ds.domain_center, (10, "Mpc"))
fn = "DD0046_sphere.h5"
full_fn = os.path.join(ytdata_dir, fn)
sphere_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, sphere_ds)
assert isinstance(sphere_ds, YTDataContainerDataset)
yield YTDataFieldTest(full_fn, ("grid", "density"))
yield YTDataFieldTest(full_fn, ("all", "particle_mass"))
cr = ds.cut_region(sphere, ['obj["temperature"] > 1e4'])
fn = "DD0046_cut_region.h5"
full_fn = os.path.join(ytdata_dir, fn)
cr_ds = data_dir_load(full_fn)
assert isinstance(cr_ds, YTDataContainerDataset)
assert (cr["temperature"] == cr_ds.data["temperature"]).all()
def test_old_nonspatial_data():
ds = data_dir_load(enzotiny)
region = ds.box([0.25] * 3, [0.75] * 3)
sphere = ds.sphere(ds.domain_center, (10, "Mpc"))
my_data = {}
my_data["region_density"] = region["density"]
my_data["sphere_density"] = sphere["density"]
fn = "test_data.h5"
full_fn = os.path.join(ytdata_dir, fn)
array_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, array_ds)
assert isinstance(array_ds, YTNonspatialDataset)
yield YTDataFieldTest(full_fn, "region_density", geometric=False)
yield YTDataFieldTest(full_fn, "sphere_density", geometric=False)
my_data = {"density": YTArray(np.linspace(1., 20., 10), "g/cm**3")}
fn = "random_data.h5"
full_fn = os.path.join(ytdata_dir, fn)
new_ds = data_dir_load(full_fn)
assert isinstance(new_ds, YTNonspatialDataset)
yield YTDataFieldTest(full_fn, "density", geometric=False)
def test_old_grid_datacontainer_data():
ds = data_dir_load(enzotiny)
fn = "DD0046_covering_grid.h5"
full_fn = os.path.join(ytdata_dir, fn)
cg_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, cg_ds)
assert isinstance(cg_ds, YTGridDataset)
yield YTDataFieldTest(full_fn, ("grid", "density"))
yield YTDataFieldTest(full_fn, ("all", "particle_mass"))
fn = "DD0046_arbitrary_grid.h5"
full_fn = os.path.join(ytdata_dir, fn)
ag_ds = data_dir_load(full_fn)
compare_unit_attributes(ds, ag_ds)
assert isinstance(ag_ds, YTGridDataset)
yield YTDataFieldTest(full_fn, ("grid", "density"))
yield YTDataFieldTest(full_fn, ("all", "particle_mass"))
my_proj = ds.proj("density", "x", weight_field="density")
frb = my_proj.to_frb(1.0, (800, 800))
fn = "DD0046_proj_frb.h5"
full_fn = os.path.join(ytdata_dir, fn)
frb_ds = data_dir_load(full_fn)
assert_allclose_units(frb["density"], frb_ds.data["density"], 1e-7)
compare_unit_attributes(ds, frb_ds)
assert isinstance(frb_ds, YTGridDataset)
yield YTDataFieldTest(full_fn, "density", geometric=False)
