/
sph_image_test.py
87 lines (62 loc) · 2.79 KB
/
sph_image_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import pynbody
import numpy as np
import numpy.testing as npt
import pylab as p
import pickle
def setup():
global f
f = pynbody.load("testdata/g15784.lr.01024")
h = f.halos()
# hard-code the centre so we're not implicitly testing the centering routine too:
cen = [0.024456279579533, -0.034112552174141, -0.122436359962132]
#cen = pynbody.analysis.halo.center(h[1],retcen=True)
#print "[%.15f, %.15f, %.15f]"%tuple(cen)
f['pos']-=cen
# derive smoothing lengths direct from file data so we are
# not testing the kdtree (which is tested elsewhere)
f.gas['smooth']=(f.gas['mass']/f.gas['rho'])**(1,3)
f.physical_units()
def test_images():
global f
im3d = pynbody.plot.sph.image(
f.gas, width=20.0, units="m_p cm^-3", noplot=True, approximate_fast=False)
im2d = pynbody.plot.sph.image(
f.gas, width=20.0, units="m_p cm^-2", noplot=True, approximate_fast=False)
im_grid = pynbody.sph.to_3d_grid(f.gas,nx=200,x2=20.0)[::50]
"""
np.save("test_im_2d.npy",im2d)
np.save("test_im_3d.npy",im3d)
np.save("test_im_grid.npy",im_grid)
"""
compare2d, compare3d = np.load("test_im_2d.npy"), np.load("test_im_3d.npy")
compare_grid = np.load("test_im_grid.npy")
npt.assert_allclose(im2d,compare2d,rtol=1e-4)
npt.assert_allclose(im3d,compare3d,rtol=1e-4)
npt.assert_allclose(im_grid,compare_grid,rtol=1e-4)
# check rectangular image is OK
im_rect = pynbody.sph.render_image(f.gas,nx=500,ny=250,x2=10.0,
approximate_fast=False).in_units("m_p cm^-3")
compare_rect = compare3d[125:-125]
npt.assert_allclose(im_rect,compare_rect,rtol=1e-4)
def test_approximate_images():
global f
im3d = pynbody.plot.sph.image(
f.gas, width=20.0, units="m_p cm^-3", noplot=True, approximate_fast=True)
im2d = pynbody.plot.sph.image(
f.gas, width=20.0, units="m_p cm^-2", noplot=True, approximate_fast=True )
compare2d, compare3d = np.load("test_im_2d.npy"), np.load("test_im_3d.npy")
# approximate interpolated images are only close in a mean sense
assert abs(np.log10(im2d/compare2d)).mean()<0.03
assert abs(np.log10(im3d/compare3d)).mean()<0.03
def test_denoise_projected_image_throws():
global f
# this should be fine:
pynbody.plot.sph.image(f.gas, width=20.0, units="m_p cm^-3", noplot=True, approximate_fast=True, denoise=True)
with npt.assert_raises(ValueError):
# this should not:
pynbody.plot.sph.image(f.gas, width=20.0, units="m_p cm^-2", noplot=True, approximate_fast=True, denoise=True)
def test_render_stars():
global f
im = pynbody.plot.stars.render(f, width=10.0, resolution=100, ret_im=True, plot=False)
compare= np.load("test_stars_2d.npy")
npt.assert_allclose(compare,im[40:60],atol=0.01)