diff options
| -rw-r--r-- | tests/test_generator_photon.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tests/test_generator_photon.py b/tests/test_generator_photon.py new file mode 100644 index 0000000..9684126 --- /dev/null +++ b/tests/test_generator_photon.py @@ -0,0 +1,20 @@ +import unittest + +import chroma.generator.photon +from chroma.generator.vertex import constant_particle_gun +from chroma.optics import water_wcsim + +class TestG4ParallelGenerator(unittest.TestCase): + def test_center(self): + '''Generate Cherenkov light at the center of the world volume''' + gen = chroma.generator.photon.G4ParallelGenerator(1, water_wcsim) + vertex = constant_particle_gun('e-', (0,0,0), (1,0,0), 100) + for event in gen.generate_events(10, vertex): + self.assertGreater(len(event.photon_start.positions), 0) + + def test_off_center(self): + '''Generate Cherenkov light at (1 m, 0 m, 0 m)''' + gen = chroma.generator.photon.G4ParallelGenerator(1, water_wcsim) + vertex = constant_particle_gun('e-', (1,0,0), (1,0,0), 100) + for event in gen.generate_events(10, vertex): + self.assertGreater(len(event.photon_start.positions), 0) |