def setUp(self):
self.num_rays = 10
dir = N.tile(N.c_[[0, 0, -1]], (1, self.num_rays))
theta = N.linspace(0, 2*N.pi, self.num_rays, endpoint=False)
position = N.vstack((N.cos(theta), N.sin(theta), N.ones(self.num_rays)))
self._bund = RayBundle()
self._bund.set_vertices(position)
self._bund.set_directions(dir)
self.gm = Paraboloid(a=5., b=5.)
self.prm = self.gm.find_intersections(N.eye(4), self._bund)
class TestInterface(unittest.TestCase):
def setUp(self):
self.num_rays = 10
dir = N.tile(N.c_[[0, 0, -1]], (1, self.num_rays))
theta = N.linspace(0, 2*N.pi, self.num_rays, endpoint=False)
position = N.vstack((N.cos(theta), N.sin(theta), N.ones(self.num_rays)))
self._bund = RayBundle()
self._bund.set_vertices(position)
self._bund.set_directions(dir)
self.gm = Paraboloid(a=5., b=5.)
self.prm = self.gm.find_intersections(N.eye(4), self._bund)
def test_find_intersections(self):
"""The correct parametric locations are found for paraboloid geometry"""
self.failUnlessEqual(self.prm.shape, (self.num_rays,))
N.testing.assert_array_almost_equal(self.prm, 0.96)
def test_get_normals(self):
"""Paraboloid surface returns center-pointing normals"""
self.gm.select_rays(N.arange(self.num_rays))
n = self.gm.get_normals() # all rays selected
N.testing.assert_array_equal(n[-1,0], n[-1,1:])
N.testing.assert_array_almost_equal(self._bund.get_vertices()[:2],
-n[:2]/N.sqrt((n[:2]**2).sum(axis=0)))
def test_inters_points_global(self):
"""Paraboloid returns correct intersections"""
self.gm.select_rays(N.arange(self.num_rays))
pts = self.gm.get_intersection_points_global()
N.testing.assert_array_equal(pts[:2], self._bund.get_vertices()[:2])
N.testing.assert_array_almost_equal(pts[2], 0.04)
class TestInterface(unittest.TestCase):
def setUp(self):
self.num_rays = 10
dir = N.tile(N.c_[[0, 0, -1]], (1, self.num_rays))
theta = N.linspace(0, 2*N.pi, self.num_rays, endpoint=False)
position = N.vstack((N.cos(theta), N.sin(theta), N.ones(self.num_rays)))
self._bund = RayBundle()
self._bund.set_vertices(position)
self._bund.set_directions(dir)
self.gm = Paraboloid(a=5., b=5.)
self.prm = self.gm.find_intersections(N.eye(4), self._bund)
def test_find_intersections(self):
"""The correct parametric locations are found for paraboloid geometry"""
self.failUnlessEqual(self.prm.shape, (self.num_rays,))
N.testing.assert_array_almost_equal(self.prm, 0.96)
def test_get_normals(self):
"""Paraboloid surface returns center-pointing normals"""
self.gm.select_rays(N.arange(self.num_rays))
n = self.gm.get_normals() # all rays selected
N.testing.assert_array_equal(n[-1,0], n[-1,1:])
N.testing.assert_array_almost_equal(self._bund.get_vertices()[:2],
-n[:2]/N.sqrt((n[:2]**2).sum(axis=0)))
def test_inters_points_global(self):
"""Paraboloid returns correct intersections"""
self.gm.select_rays(N.arange(self.num_rays))
pts = self.gm.get_intersection_points_global()
N.testing.assert_array_equal(pts[:2], self._bund.get_vertices()[:2])
N.testing.assert_array_almost_equal(pts[2], 0.04)
def setUp(self):
self.num_rays = 10
dir = N.tile(N.c_[[0, 0, -1]], (1, self.num_rays))
theta = N.linspace(0, 2*N.pi, self.num_rays, endpoint=False)
position = N.vstack((N.cos(theta), N.sin(theta), N.ones(self.num_rays)))
self._bund = RayBundle()
self._bund.set_vertices(position)
self._bund.set_directions(dir)
self.gm = Paraboloid(a=5., b=5.)
self.prm = self.gm.find_intersections(N.eye(4), self._bund)
def setUp(self):
self.assembly = Assembly()
surface1 = Surface(Paraboloid(), optics_callables.perfect_mirror)
self.object = AssembledObject()
self.object.add_surface(surface1)
self.assembly.add_object(self.object)
x = 1. / (math.sqrt(2))
dir = N.c_[[0, 0, -1.], [0, x, -x]]
position = N.c_[[0, 0, 1.], [0, 0, 1.]]
self._bund = RayBundle(position,
dir,
energy=N.ones(2),
ref_index=N.ones(2))
