diff options
| -rw-r--r-- | test/test_rayleigh.py | 5 | ||||
| -rw-r--r-- | test/test_reemission.py | 2 |
2 files changed, 4 insertions, 3 deletions
diff --git a/test/test_rayleigh.py b/test/test_rayleigh.py index 7657fc1..0b3274d 100644 --- a/test/test_rayleigh.py +++ b/test/test_rayleigh.py @@ -2,6 +2,7 @@ import unittest import numpy as np from chroma.geometry import Solid, Geometry +from chroma.loader import create_geometry_from_obj from chroma.make import box from chroma.sim import Simulation from chroma.demo.optics import water @@ -15,8 +16,8 @@ class TestRayleigh(unittest.TestCase): def setUp(self): self.cube = Geometry(water) self.cube.add_solid(Solid(box(100,100,100), water, water)) - self.cube.build(use_cache=False) - self.sim = Simulation(self.cube, geant4_processes=0) + self.geo = create_geometry_from_obj(self.cube, update_bvh_cache=False) + self.sim = Simulation(self.geo, geant4_processes=0) nphotons = 100000 pos = np.tile([0,0,0], (nphotons,1)).astype(np.float32) diff --git a/test/test_reemission.py b/test/test_reemission.py index c6854be..8fd3569 100644 --- a/test/test_reemission.py +++ b/test/test_reemission.py @@ -40,7 +40,7 @@ class TestReemission(unittest.TestCase): world = Geometry(vacuum) world.add_solid(Solid(sphere(1000), vacuum, vacuum, surface=detector)) world.add_solid(Solid(sphere(100), scint, vacuum)) - w = create_geometry_from_obj(world) + w = create_geometry_from_obj(world, update_bvh_cache=False) sim = Simulation(w, geant4_processes=0) |