From 142b3c3caff164deb9bc7b2848e58e52387723ff Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Fri, 16 Sep 2011 15:02:02 -0400 Subject: Move CUDA source inside chroma package, rename tests directory to test --- test/test_generator_photon.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 test/test_generator_photon.py (limited to 'test/test_generator_photon.py') diff --git a/test/test_generator_photon.py b/test/test_generator_photon.py new file mode 100644 index 0000000..13501fe --- /dev/null +++ b/test/test_generator_photon.py @@ -0,0 +1,21 @@ +import unittest +import itertools + +from chroma import generator +from chroma.generator.vertex import constant_particle_gun +from chroma import optics + +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) + 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) + 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