class TestModule(TestCase):
GEO1 = ['MODULE ( 1) Test',
'MATERIAL( 1)',
'SURFACE ( 1), SIDE POINTER=(-1)',
'SURFACE ( 2), SIDE POINTER=( 1)',
'MODULE ( 2)',
'1111111111111111111111111111111111111111111111111111111111111111',
' OMEGA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' THETA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' PHI=(+1.800000000000000E+02, 0) DEG (DEFAULT=0.0)',
'X-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Y-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Z-SHIFT=(-1.000000000000000E+05, 0) (DEFAULT=0.0)']
GEO2 = ['MODULE ( 2) ',
'MATERIAL( 0)',
'1111111111111111111111111111111111111111111111111111111111111111',
' OMEGA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' THETA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' PHI=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
'X-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Y-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Z-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)']
def setUp(self):
TestCase.setUp(self)
intforce = InteractionForcing(ELECTRON, HARD_BREMSSTRAHLUNG_EMISSION, -4)
mat1 = PenelopeMaterial.pure(29, interaction_forcings=[intforce],
maximum_step_length_m=1e4)
mat1._index = 1
mat2 = VACUUM
mat2._index = 0
surface1 = SurfaceImplicit()
surface1._index = 0
surface2 = SurfaceImplicit()
surface2._index = 1
self.module2 = Module(None, mat2)
self.module2._index = 1
self.module1 = Module(None, mat1, 'Test')
self.module1.add_surface(surface1, -1)
self.module1.add_surface(surface2, 1)
self.module1.add_module(self.module2)
self.module1.rotation.phi_rad = radians(180)
self.module1.shift.z_m = -1e3
self.module1._index = 0
self.materials_lookup = {0: mat2, 1: mat1}
self.surfaces_lookup = {0: surface1, 1: surface2}
self.modules_lookup = {0: self.module1, 1: self.module2}
def tearDown(self):
TestCase.tearDown(self)
def testskeleton(self):
# Module 1
self.assertEqual('Copper', str(self.module1.material))
self.assertAlmostEqual(1e4, self.module1.material.maximum_step_length_m, 4)
self.assertEqual(1, len(self.module1.material.interaction_forcings))
self.assertEqual('Test', self.module1.description)
self.assertAlmostEqual(radians(180), self.module1.rotation.phi_rad, 4)
self.assertAlmostEqual(-1e3, self.module1.shift.z_m, 4)
self.assertEqual(2, len(self.module1.get_surfaces()))
self.assertEqual(1, len(self.module1.get_modules()))
# Module 2
self.assertEqual(str(VACUUM), str(self.module2.material))
self.assertEqual(0, len(self.module2.get_surfaces()))
self.assertEqual(0, len(self.module2.get_modules()))
def testto_geo(self):
# Module 1
lines = self.module1.to_geo()
self.assertEqual(12, len(lines))
self.assertEqual(self.GEO1, lines)
# Module 2
lines = self.module2.to_geo()
self.assertEqual(9, len(lines))
self.assertEqual(self.GEO2, lines)
class TestModule(TestCase):
GEO1 = [
'MODULE ( 1) Test', 'MATERIAL( 1)',
'SURFACE ( 1), SIDE POINTER=(-1)',
'SURFACE ( 2), SIDE POINTER=( 1)', 'MODULE ( 2)',
'1111111111111111111111111111111111111111111111111111111111111111',
' OMEGA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' THETA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' PHI=(+1.800000000000000E+02, 0) DEG (DEFAULT=0.0)',
'X-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Y-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Z-SHIFT=(-1.000000000000000E+05, 0) (DEFAULT=0.0)'
]
GEO2 = [
'MODULE ( 2) ', 'MATERIAL( 0)',
'1111111111111111111111111111111111111111111111111111111111111111',
' OMEGA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' THETA=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
' PHI=(+0.000000000000000E+00, 0) DEG (DEFAULT=0.0)',
'X-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Y-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)',
'Z-SHIFT=(+0.000000000000000E+00, 0) (DEFAULT=0.0)'
]
def setUp(self):
TestCase.setUp(self)
intforce = InteractionForcing(ELECTRON, HARD_BREMSSTRAHLUNG_EMISSION,
-4)
mat1 = PenelopeMaterial.pure(29,
interaction_forcings=[intforce],
maximum_step_length_m=1e4)
mat1._index = 1
mat2 = VACUUM
mat2._index = 0
surface1 = SurfaceImplicit()
surface1._index = 0
surface2 = SurfaceImplicit()
surface2._index = 1
self.module2 = Module(None, mat2)
self.module2._index = 1
self.module1 = Module(None, mat1, 'Test')
self.module1.add_surface(surface1, -1)
self.module1.add_surface(surface2, 1)
self.module1.add_module(self.module2)
self.module1.rotation.phi_rad = radians(180)
self.module1.shift.z_m = -1e3
self.module1._index = 0
self.materials_lookup = {0: mat2, 1: mat1}
self.surfaces_lookup = {0: surface1, 1: surface2}
self.modules_lookup = {0: self.module1, 1: self.module2}
def tearDown(self):
TestCase.tearDown(self)
def testskeleton(self):
# Module 1
self.assertEqual('Copper', str(self.module1.material))
self.assertAlmostEqual(1e4,
self.module1.material.maximum_step_length_m, 4)
self.assertEqual(1, len(self.module1.material.interaction_forcings))
self.assertEqual('Test', self.module1.description)
self.assertAlmostEqual(radians(180), self.module1.rotation.phi_rad, 4)
self.assertAlmostEqual(-1e3, self.module1.shift.z_m, 4)
self.assertEqual(2, len(self.module1.get_surfaces()))
self.assertEqual(1, len(self.module1.get_modules()))
# Module 2
self.assertEqual(str(VACUUM), str(self.module2.material))
self.assertEqual(0, len(self.module2.get_surfaces()))
self.assertEqual(0, len(self.module2.get_modules()))
def testto_geo(self):
# Module 1
lines = self.module1.to_geo()
self.assertEqual(12, len(lines))
self.assertEqual(self.GEO1, lines)
# Module 2
lines = self.module2.to_geo()
self.assertEqual(9, len(lines))
self.assertEqual(self.GEO2, lines)
