add ray_intersection unit test.

4 files changed, 28 insertions, 1 deletions

diff --git a/project.py b/project.py
index 10254f2..7d9cabe 100644
--- a/project.py
+++ b/project.py
@@ -2,7 +2,7 @@ import numpy as np
 from itertools import repeat
 from chroma.transform import rotate, normalize
 
-def from_film(position, axis1=(0,0,1), axis2=(1,0,0), size=(800,600),
+def from_film(position=(0,0,0), axis1=(0,0,1), axis2=(1,0,0), size=(800,600),
               width=0.035, focal_length=0.018):
     """Project rays from a piece of film located at whose focal point is
     located at `position`. `axis1` and `axis2` specify the vectors pointing
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/data/ray_intersection.npz b/tests/data/ray_intersection.npz
new file mode 100644
index 0000000..fba0e40
--- /dev/null
+++ b/tests/data/ray_intersection.npz
diff --git a/tests/test_ray_intersection.py b/tests/test_ray_intersection.py
new file mode 100644
index 0000000..7d0c53c
--- /dev/null
+++ b/tests/test_ray_intersection.py
@@ -0,0 +1,27 @@
+import unittest
+import chroma
+import numpy as np
+import os
+from pycuda import gpuarray as ga
+
+class TestRayIntersection(unittest.TestCase):
+    def setUp(self):
+        self.context = chroma.gpu.create_cuda_context()
+        self.module = chroma.gpu.get_cu_module('mesh.h')
+        self.gpu_funcs = chroma.gpu.GPUFuncs(self.module)
+        self.box = chroma.gpu.GPUGeometry(chroma.build(chroma.make.cube()))
+
+        pos, dir = chroma.project.from_film()
+        self.pos_gpu = ga.to_gpu(chroma.gpu.to_float3(pos))
+        self.dir_gpu = ga.to_gpu(chroma.gpu.to_float3(dir))
+
+        testdir = os.path.dirname(os.path.abspath(chroma.tests.__file__))
+        self.dx_standard = np.load(os.path.join(testdir,
+                                                'data/ray_intersection.npz'))
+    def test_intersection_distance(self):
+        dx = ga.zeros(self.pos_gpu.size, dtype=np.float32)
+        self.gpu_funcs.distance_to_mesh(np.int32(self.pos_gpu.size), self.pos_gpu, self.dir_gpu, self.box.gpudata, dx, block=(64,1,1), grid=(self.pos_gpu.size//64+1,1))
+        self.assertTrue((dx.get() == self.dx_standard).all())
+
+    def tearDown(self):
+        self.context.pop()