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 --- tests/test_generator_vertex.py | 23 ----------------------- 1 file changed, 23 deletions(-) delete mode 100644 tests/test_generator_vertex.py (limited to 'tests/test_generator_vertex.py') diff --git a/tests/test_generator_vertex.py b/tests/test_generator_vertex.py deleted file mode 100644 index cec363d..0000000 --- a/tests/test_generator_vertex.py +++ /dev/null @@ -1,23 +0,0 @@ -import unittest -import itertools -import numpy as np -import chroma.generator.vertex - -class TestParticleGun(unittest.TestCase): - def test_constant_particle_gun_center(self): - '''Generate electron vertices at the center of the world volume.''' - vertex = chroma.generator.vertex.constant_particle_gun('e-', (0,0,0), (1,0,0), 100) - for ev in itertools.islice(vertex, 100): - self.assertEquals(ev.primary_vertex.particle_name, 'e-') - self.assertTrue(np.allclose(ev.primary_vertex.pos, [0,0,0])) - self.assertTrue(np.allclose(ev.primary_vertex.dir, [1,0,0])) - self.assertTrue(np.allclose(ev.primary_vertex.ke, 100)) - - def test_off_center(self): - '''Generate electron vertices at (1,0,0) in the world volume.''' - vertex = chroma.generator.vertex.constant_particle_gun('e-', (1,0,0), (1,0,0), 100) - for ev in itertools.islice(vertex, 100): - self.assertEquals(ev.primary_vertex.particle_name, 'e-') - self.assertTrue(np.allclose(ev.primary_vertex.pos, [1,0,0])) - self.assertTrue(np.allclose(ev.primary_vertex.dir, [1,0,0])) - self.assertTrue(np.allclose(ev.primary_vertex.ke, 100)) -- cgit