diff options
Diffstat (limited to 'test/test_propagation.py')
-rw-r--r-- | test/test_propagation.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/test_propagation.py b/test/test_propagation.py index e1b52ae..114fcfb 100644 --- a/test/test_propagation.py +++ b/test/test_propagation.py @@ -2,6 +2,7 @@ import unittest import numpy as np from chroma.geometry import Solid, Geometry, vacuum +from chroma.loader import create_geometry_from_obj from chroma.make import box from chroma.sim import Simulation from chroma.event import Photons @@ -19,8 +20,8 @@ class TestPropagation(unittest.TestCase): # Setup geometry cube = Geometry(vacuum) cube.add_solid(Solid(box(100,100,100), vacuum, vacuum)) - cube.build(use_cache=False) - sim = Simulation(cube, geant4_processes=0) + geo = create_geometry_from_obj(cube, update_bvh_cache=False) + sim = Simulation(geo, geant4_processes=0) # Create initial photons nphotons = 10000 |