def test_ahf_values():
h_id, ngas, nstar, ndm, ntot, rvir = stat.HaloStatFile(ts1.filename).read(
"n_gas", "n_star", "n_dm", "npart", "Rvir")
assert all(h_id == [1, 2, 3, 4])
assert all(ngas == [324272, 47634, 53939, 19920])
assert all(nstar == [1227695, 55825, 24561, 7531])
assert all(ntot == [5900575, 506026, 498433, 226976])
assert all(ndm == [4348608, 402567, 419933, 199525])
npt.assert_allclose(rvir, [195.87, 88.75, 90.01, 69.41])
def test_idl_values():
h_id, ntot, mvir = stat.HaloStatFile(ts2.filename).read("npart", "Mvir")
assert all(h_id == [1, 49, 52, 58, 94, 121, 127, 148, 163])
assert all(
ntot == [3273314, 27631, 24654, 22366, 12915, 9831, 9498, 8200, 7256])
npt.assert_almost_equal(mvir, [
1.12282280e+12, 1.19939950e+10, 1.19538740e+10, 3.07825010e+10,
1.76325820e+10, 1.33353700e+10, 1.28836660e+10, 1.11229900e+10,
9.84248220e+09
],
decimal=5)
def test_statfile_identity():
global ts1, ts2
assert isinstance(stat.HaloStatFile(ts1.filename), stat.AHFStatFile)
assert isinstance(stat.HaloStatFile(ts2.filename), stat.AmigaIDLStatFile)