class TestParabolicDish(unittest.TestCase):
def setUp(self):
pos = N.zeros((3, 4))
pos[0] = N.r_[0, 0.5, 2, -2]
pos[2] = 2.
dir = N.tile(N.c_[[0, 0, -1]], (1, 4))
self.bund = RayBundle()
self.bund.set_vertices(pos)
self.bund.set_directions(dir)
self.surf = Surface(ParabolicDishGM(2., 1.), opt.perfect_mirror)
def test_selection_at_origin(self):
"""Simple dish rejects missing rays"""
misses = N.isinf(self.surf.register_incoming(self.bund))
N.testing.assert_array_equal(misses, N.r_[False, False, True, True])
def test_transformed(self):
"""Translated and rotated dish rejects missing rays"""
trans = generate_transform(N.r_[1., 0., 0.], N.pi / 4.,
N.c_[[0., 0., 1.]])
self.surf.transform_frame(trans)
misses = N.isinf(self.surf.register_incoming(self.bund))
N.testing.assert_array_equal(misses, N.r_[False, False, True, True])
def test_mesh(self):
"""Parabolic dish mesh looks OK"""
p = ParabolicDishGM(5, 3)
x, y, z = p.mesh(5)
N.testing.assert_array_almost_equal(z, p.a * (x**2 + y**2))
self.failIf(N.any(x**2 + y**2 > 6.25))
class TestParabolicDish(unittest.TestCase):
def setUp(self):
pos = N.zeros((3,4))
pos[0] = N.r_[0, 0.5, 2, -2]
pos[2] = 2.
dir = N.tile(N.c_[[0,0,-1]], (1,4))
self.bund = RayBundle()
self.bund.set_vertices(pos)
self.bund.set_directions(dir)
self.surf = Surface(ParabolicDishGM(2., 1.), opt.perfect_mirror)
def test_selection_at_origin(self):
"""Simple dish rejects missing rays"""
misses = N.isinf(self.surf.register_incoming(self.bund))
N.testing.assert_array_equal(misses, N.r_[False, False, True, True])
def test_transformed(self):
"""Translated and rotated dish rejects missing rays"""
trans = generate_transform(N.r_[1., 0., 0.], N.pi/4., N.c_[[0., 0., 1.]])
self.surf.transform_frame(trans)
misses = N.isinf(self.surf.register_incoming(self.bund))
N.testing.assert_array_equal(misses, N.r_[False, False, True, True])
def test_mesh(self):
"""Parabolic dish mesh looks OK"""
p = ParabolicDishGM(5, 3)
x, y, z = p.mesh(5)
N.testing.assert_array_almost_equal(z, p.a*(x**2 + y**2))
self.failIf(N.any(x**2 + y**2 > 6.25))
