New demo detector for Chroma that is not dependent on any internal experiment information.

1 files changed, 3 insertions, 3 deletions

diff --git a/test/test_generator_photon.py b/test/test_generator_photon.py
index 13501fe..6952da9 100644
--- a/test/test_generator_photon.py
+++ b/test/test_generator_photon.py
@@ -3,19 +3,19 @@ import itertools
 
 from chroma import generator
 from chroma.generator.vertex import constant_particle_gun
-from chroma import optics
+from chroma.demo.optics import water
 
 class TestG4ParallelGenerator(unittest.TestCase):
     def test_center(self):
         '''Generate Cherenkov light at the center of the world volume'''
-        gen = generator.photon.G4ParallelGenerator(1, optics.water_wcsim)
+        gen = generator.photon.G4ParallelGenerator(1, water)
         vertex = itertools.islice(constant_particle_gun('e-', (0,0,0), (1,0,0), 100), 10)
         for event in gen.generate_events(vertex):
             self.assertGreater(len(event.photons_beg.pos), 0)
 
     def test_off_center(self):
         '''Generate Cherenkov light at (1 m, 0 m, 0 m)'''
-        gen = generator.photon.G4ParallelGenerator(1, optics.water_wcsim)
+        gen = generator.photon.G4ParallelGenerator(1, water)
         vertex = itertools.islice(constant_particle_gun('e-', (1,0,0), (1,0,0), 100), 10)
         for event in gen.generate_events(vertex):
             self.assertGreater(len(event.photons_beg.pos), 0)