def setUp(self):
TestCase.setUp(self)
self.testdata = os.path.join(os.path.dirname(__file__), 'testdata',
'test1')
self.i = Importer()
def setUp(self):
TestCase.setUp(self)
self.testdata = os.path.join(os.path.dirname(__file__),
'testdata', 'test1')
self.i = Importer()
class TestImporter(TestCase):
def setUp(self):
TestCase.setUp(self)
self.testdata = os.path.join(os.path.dirname(__file__),
'testdata', 'test1')
self.i = Importer()
def tearDown(self):
TestCase.tearDown(self)
def testskeleton(self):
self.assertTrue(True)
def test_detector_trajectory(self):
# Create
ops = Options(name='test1')
ops.beam.energy_eV = 20e3
ops.detectors['trajectories'] = TrajectoryDetector(50)
# Import
resultcontainer = self.i.import_(ops, self.testdata)
# Test
result = resultcontainer['trajectories']
self.assertEqual(559, len(result))
trajectory = list(result)[0]
self.assertTrue(trajectory.is_primary())
self.assertFalse(trajectory.is_secondary())
self.assertIs(ELECTRON, trajectory.particle)
self.assertIs(NO_COLLISION, trajectory.collision)
self.assertEqual(EXIT_STATE_ABSORBED, trajectory.exit_state)
self.assertEqual(577, len(trajectory.interactions))
self.assertEqual(5, trajectory.interactions.shape[1])
class TestImporter(TestCase):
def setUp(self):
TestCase.setUp(self)
self.testdata = os.path.join(os.path.dirname(__file__), 'testdata',
'test1')
self.i = Importer()
def tearDown(self):
TestCase.tearDown(self)
def testskeleton(self):
self.assertTrue(True)
def test_detector_trajectory(self):
# Create
ops = Options(name='test1')
ops.beam.energy_eV = 20e3
ops.detectors['trajectories'] = TrajectoryDetector(50)
# Import
resultcontainer = self.i.import_(ops, self.testdata)
# Test
result = resultcontainer['trajectories']
self.assertEqual(559, len(result))
trajectory = list(result)[0]
self.assertTrue(trajectory.is_primary())
self.assertFalse(trajectory.is_secondary())
self.assertIs(ELECTRON, trajectory.particle)
self.assertIs(NO_COLLISION, trajectory.collision)
self.assertEqual(EXIT_STATE_ABSORBED, trajectory.exit_state)
self.assertEqual(577, len(trajectory.interactions))
self.assertEqual(5, trajectory.interactions.shape[1])
