def test_on_off_compare():
# fake density field that varies in the x-direction only
den = np.arange(32**3) / 32**2 + 1
den = den.reshape(32, 32, 32)
den = np.array(den, dtype=np.float64)
data = dict(density=(den, "g/cm**3"))
bbox = np.array([[-1.5, 1.5], [-1.5, 1.5], [-1.5, 1.5]])
ds = load_uniform_grid(data,
den.shape,
length_unit="Mpc",
bbox=bbox,
nprocs=64)
sl_on = SlicePlot(ds, "z", [("gas", "density")])
L = [0, 0, 1]
north_vector = [0, 1, 0]
sl_off = OffAxisSlicePlot(ds,
L, ("gas", "density"),
center=[0, 0, 0],
north_vector=north_vector)
assert_array_almost_equal(sl_on.frb[("gas", "density")],
sl_off.frb[("gas", "density")])
sl_on.set_buff_size((800, 400))
sl_on._recreate_frb()
sl_off.set_buff_size((800, 400))
sl_off._recreate_frb()
assert_array_almost_equal(sl_on.frb[("gas", "density")],
sl_off.frb[("gas", "density")])
def test_frb_regen():
ds = fake_random_ds(32)
slc = SlicePlot(ds, 2, ("gas", "density"))
slc.set_buff_size(1200)
assert_equal(slc.frb[("gas", "density")].shape, (1200, 1200))
slc.set_buff_size((400.0, 200.7))
assert_equal(slc.frb[("gas", "density")].shape, (200, 400))
def test_set_unit():
ds = fake_random_ds(32, fields=(("gas", "temperature"),), units=("K",))
slc = SlicePlot(ds, 2, ("gas", "temperature"))
orig_array = slc.frb["gas", "temperature"].copy()
slc.set_unit(("gas", "temperature"), "degF")
assert str(slc.frb["gas", "temperature"].units) == "°F"
assert_array_almost_equal(
np.array(slc.frb["gas", "temperature"]), np.array(orig_array) * 1.8 - 459.67
)
# test that a plot modifying function that destroys the frb preserves the
# new unit
slc.set_buff_size(1000)
assert str(slc.frb["gas", "temperature"].units) == "°F"
slc.set_buff_size(800)
slc.set_unit(("gas", "temperature"), "K")
assert str(slc.frb["gas", "temperature"].units) == "K"
assert_array_almost_equal(slc.frb["gas", "temperature"], orig_array)
slc.set_unit(("gas", "temperature"), "keV", equivalency="thermal")
assert str(slc.frb["gas", "temperature"].units) == "keV"
assert_array_almost_equal(
slc.frb["gas", "temperature"], (orig_array * kboltz).to("keV")
)
# test that a plot modifying function that destroys the frb preserves the
# new unit with an equivalency
slc.set_buff_size(1000)
assert str(slc.frb["gas", "temperature"].units) == "keV"
# test that destroying the FRB then changing the unit using an equivalency
# doesn't error out, see issue #1316
slc = SlicePlot(ds, 2, ("gas", "temperature"))
slc.set_buff_size(1000)
slc.set_unit(("gas", "temperature"), "keV", equivalency="thermal")
assert str(slc.frb["gas", "temperature"].units) == "keV"