diff options
| author | Stan Seibert <stan@mtrr.org> | 2011-09-03 09:21:36 -0400 |
|---|---|---|
| committer | Stan Seibert <stan@mtrr.org> | 2011-09-03 09:21:36 -0400 |
| commit | 38f05bf761490def1886016524f328528b08f549 (patch) | |
| tree | e0ee6555ee8bdf02a9e0b832a33707bcee06a3fa /tests/test_generator_vertex.py | |
| parent | 48550062440c5b7f1479ecbe17fd4b024a90fca2 (diff) | |
| parent | 707ca1b366f11032682cc864ca2848905e6b485c (diff) | |
| download | chroma-38f05bf761490def1886016524f328528b08f549.tar.gz chroma-38f05bf761490def1886016524f328528b08f549.tar.bz2 chroma-38f05bf761490def1886016524f328528b08f549.zip | |
merge
Diffstat (limited to 'tests/test_generator_vertex.py')
| -rw-r--r-- | tests/test_generator_vertex.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/tests/test_generator_vertex.py b/tests/test_generator_vertex.py index d773579..cec363d 100644 --- a/tests/test_generator_vertex.py +++ b/tests/test_generator_vertex.py @@ -8,16 +8,16 @@ class TestParticleGun(unittest.TestCase): '''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.particle_name, 'e-') - self.assertTrue(np.allclose(ev.gen_position, [0,0,0])) - self.assertTrue(np.allclose(ev.gen_direction, [1,0,0])) - self.assertTrue(np.allclose(ev.gen_total_energy, 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.particle_name, 'e-') - self.assertTrue(np.allclose(ev.gen_position, [1,0,0])) - self.assertTrue(np.allclose(ev.gen_direction, [1,0,0])) - self.assertTrue(np.allclose(ev.gen_total_energy, 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)) |