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)