def test_intersect(self):
sphere1 = Sphere(Vector3f(2.0, 0.0, 0.0), 1)
sphere2 = Sphere(Vector3f(-2.0, 0.0, 0.0), 1)
self.assertEqual(sphere1.intersect(Vector3f(0, 0, 0), Vector3f(1, 0, 0)), 1.0)
self.assertEqual(sphere1.intersect(Vector3f(0, 0, 0), Vector3f(0, 1, 0)), "NONE")
self.assertEqual(sphere2.intersect(Vector3f(0, 0, 0), Vector3f(1, 0, 0)), "NONE")
class TestSphere(TestRenderablePrimitive):
def setUp(self):
self.renderable = Sphere()
self.renderable2 = Sphere()
def test_interface(self):
super(TestSphere, self).test_interface()
for func in ['setRadius']:
self.assertTrue(hasattr(self.renderable, func) and callable(getattr(self.renderable, func)), "Interface requires function: " + func)
def test_intersect(self):
sphere1 = Sphere(Vector3f(2.0, 0.0, 0.0), 1)
sphere2 = Sphere(Vector3f(-2.0, 0.0, 0.0), 1)
self.assertEqual(sphere1.intersect(Vector3f(0, 0, 0), Vector3f(1, 0, 0)), 1.0)
self.assertEqual(sphere1.intersect(Vector3f(0, 0, 0), Vector3f(0, 1, 0)), "NONE")
self.assertEqual(sphere2.intersect(Vector3f(0, 0, 0), Vector3f(1, 0, 0)), "NONE")
def test_normalAt(self):
sphere1 = Sphere(Vector3f(2.0, 0.0, 0.0), 1)
self.assertEqual(sphere1.normalAt(Vector3f(1, 0, 0)).toList(), [-1, 0, 0])
self.assertEqual(sphere1.normalAt(Vector3f(1, 0.1, 0)).toList(), Vector3f(-1, 0.1, 0.0).toList())
self.assertEqual(sphere1.normalAt(Vector3f(5, 0, 0)).toList(), [3, 0, 0])
def test_setRadius(self):
self.assertEqual(self.renderable.setRadius(1.0), 1.0)
self.assertEqual(self.renderable.setRadius(0.0), 1.0)
self.assertEqual(self.renderable.setRadius(-1.0), 1.0)
self.assertEqual(self.renderable.setRadius(1.1), 1.1)
self.assertEqual(self.renderable.setRadius(0.0005), 0.0005)
self.assertEqual(self.renderable.setRadius(), 0.0005)
self.assertEqual(self.renderable.setRadius(1), 1.0)
def test_integration_renderComplexScene(self):
self.raytracer.setOutput('test_integration_renderComplexScene.png', 300, 300)
self.raytracer.setAmbient([0.2, 0.2, 0.2])
sph1 = Sphere(Vector3f(0.0, 1.0, 0.0), 0.5)
sph2 = Sphere(Vector3f(-1.0, 1.0, 0.0), 0.4)
sph3 = Sphere(Vector3f(0.2, -1.0, 1.2), 0.5)
sph4 = Sphere(Vector3f(1.0, -1.0, -1.0), 0.3)
sph5 = Sphere(Vector3f(-1.0, -1.0, -1.0), 0.3)
sph1.setDiffuseColor([0.8, 0.0, 0.0])
sph2.setDiffuseColor([1.0, 1.0, 1.0])
sph2.setDiffuseConstant(0.5)
sph3.setDiffuseColor([0.0, 0.0, 0.8])
sph3.setShininess(20)
sph4.setSpecularColor([0.0, 1.0, 0.0])
sph5.setDiffuseColor([1.0, 0.0, 1.0])
sph5.setShininess(200)
self.raytracer.addPrimitive(sph1)
self.raytracer.addPrimitive(sph2)
self.raytracer.addPrimitive(sph3)
self.raytracer.addPrimitive(sph4)
self.raytracer.addPrimitive(sph5)
self.raytracer.addPointLight(Vector3f(1.0,-4.0,1.0), [1.0, 1.0, 1.0], 2)
self.raytracer.setCamera(Vector3f(0.0, -6.0, 0.0), Vector3f(0.0, 1.0, 0.0), Vector3f(0.0, 0.0, 1.0), 70)
start = time()
self.raytracer.render()
end = time()
self.assertTrue(os.path.exists('./test_integration_renderComplexScene.png'), "The file wasn't created upon rendering")
print "Rendered a 300 * 300 image with 4 spheres and 1 light in {} seconds".format(end - start)
def setUp(self):
self.renderable = Sphere()
self.renderable2 = Sphere()
def test_normalAt(self):
sphere1 = Sphere(Vector3f(2.0, 0.0, 0.0), 1)
self.assertEqual(sphere1.normalAt(Vector3f(1, 0, 0)).toList(), [-1, 0, 0])
self.assertEqual(sphere1.normalAt(Vector3f(1, 0.1, 0)).toList(), Vector3f(-1, 0.1, 0.0).toList())
self.assertEqual(sphere1.normalAt(Vector3f(5, 0, 0)).toList(), [3, 0, 0])