From b5a0b1e36543e737c522f69e8c3e8a18729de429 Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Thu, 29 Sep 2011 17:10:16 -0400 Subject: New demo detector for Chroma that is not dependent on any internal experiment information. --- test/test_generator_photon.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test/test_generator_photon.py') 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) -- cgit