diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/data/ray_intersection.npy | bin | 0 -> 1920080 bytes | |||
| -rw-r--r-- | test/data/ray_intersection.npz | bin | 1920080 -> 0 bytes | |||
| -rw-r--r-- | test/test_ray_intersection.py | 6 |
3 files changed, 4 insertions, 2 deletions
diff --git a/test/data/ray_intersection.npy b/test/data/ray_intersection.npy Binary files differnew file mode 100644 index 0000000..43f51c2 --- /dev/null +++ b/test/data/ray_intersection.npy diff --git a/test/data/ray_intersection.npz b/test/data/ray_intersection.npz Binary files differdeleted file mode 100644 index fba0e40..0000000 --- a/test/data/ray_intersection.npz +++ /dev/null diff --git a/test/test_ray_intersection.py b/test/test_ray_intersection.py index ce364e0..8a33ac3 100644 --- a/test/test_ray_intersection.py +++ b/test/test_ray_intersection.py @@ -9,7 +9,7 @@ class TestRayIntersection(unittest.TestCase): 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())) + self.box = chroma.gpu.GPUGeometry(chroma.build(chroma.make.cube(size=1000.0))) pos, dir = chroma.project.from_film() self.pos_gpu = ga.to_gpu(chroma.gpu.to_float3(pos)) @@ -17,10 +17,12 @@ class TestRayIntersection(unittest.TestCase): testdir = os.path.dirname(os.path.abspath(__file__)) self.dx_standard = np.load(os.path.join(testdir, - 'data/ray_intersection.npz')) + 'data/ray_intersection.npy')) + 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): |