def setUp(self):
unittest.TestCase.setUp(self)
self.h = HorizontalLayersXMLHandler()
self.obj1 = HorizontalLayers(Material.pure(29), None, 1.1, 2.2)
self.obj1.add_layer(Material.pure(30), 123.456)
self.obj1.add_layer(Material.pure(31), 456.789)
self.obj2 = HorizontalLayers()
self.obj2.add_layer(Material.pure(30), 123.456)
self.obj2.add_layer(VACUUM, 50.0)
self.obj2.add_layer(Material.pure(31), 456.789)
etree.register_namespace('mc', 'http://pymontecarlo.sf.net')
source = BytesIO(b'<mc:horizontalLayers xmlns:mc="http://pymontecarlo.sf.net" rotation="2.2" tilt="1.1"><materials><mc:material _index="1" density="7140.0" name="Zinc"><composition><element weightFraction="1.0" z="30" /></composition></mc:material><mc:material _index="2" density="5910.0" name="Gallium"><composition><element weightFraction="1.0" z="31" /></composition></mc:material><mc:material _index="3" density="8960.0" name="Copper"><composition><element weightFraction="1.0" z="29" /></composition></mc:material></materials><substrate material="3" /><layers><layer material="1" thickness="123.456" /><layer material="2" thickness="456.789" /></layers></mc:horizontalLayers>')
self.element1 = etree.parse(source).getroot()
etree.register_namespace('mc', 'http://pymontecarlo.sf.net')
source = BytesIO(b'<mc:horizontalLayers xmlns:mc="http://pymontecarlo.sf.net" rotation="0.0" tilt="0.0"><materials><mc:material _index="1" density="7140.0" name="Zinc"><composition><element weightFraction="1.0" z="30" /></composition></mc:material><mc:material _index="2" density="5910.0" name="Gallium"><composition><element weightFraction="1.0" z="31" /></composition></mc:material></materials><layers><layer material="1" thickness="123.456" /><layer material="0" thickness="50.0" /><layer material="2" thickness="456.789" /></layers></mc:horizontalLayers>')
self.element2 = etree.parse(source).getroot()
class TestHorizontalLayersXMLHandler(unittest.TestCase):
def setUp(self):
unittest.TestCase.setUp(self)
self.h = HorizontalLayersXMLHandler()
self.obj1 = HorizontalLayers(Material.pure(29), None, 1.1, 2.2)
self.obj1.add_layer(Material.pure(30), 123.456)
self.obj1.add_layer(Material.pure(31), 456.789)
self.obj2 = HorizontalLayers()
self.obj2.add_layer(Material.pure(30), 123.456)
self.obj2.add_layer(VACUUM, 50.0)
self.obj2.add_layer(Material.pure(31), 456.789)
etree.register_namespace('mc', 'http://pymontecarlo.sf.net')
source = BytesIO(b'<mc:horizontalLayers xmlns:mc="http://pymontecarlo.sf.net" rotation="2.2" tilt="1.1"><materials><mc:material _index="1" density="7140.0" name="Zinc"><composition><element weightFraction="1.0" z="30" /></composition></mc:material><mc:material _index="2" density="5910.0" name="Gallium"><composition><element weightFraction="1.0" z="31" /></composition></mc:material><mc:material _index="3" density="8960.0" name="Copper"><composition><element weightFraction="1.0" z="29" /></composition></mc:material></materials><substrate material="3" /><layers><layer material="1" thickness="123.456" /><layer material="2" thickness="456.789" /></layers></mc:horizontalLayers>')
self.element1 = etree.parse(source).getroot()
etree.register_namespace('mc', 'http://pymontecarlo.sf.net')
source = BytesIO(b'<mc:horizontalLayers xmlns:mc="http://pymontecarlo.sf.net" rotation="0.0" tilt="0.0"><materials><mc:material _index="1" density="7140.0" name="Zinc"><composition><element weightFraction="1.0" z="30" /></composition></mc:material><mc:material _index="2" density="5910.0" name="Gallium"><composition><element weightFraction="1.0" z="31" /></composition></mc:material></materials><layers><layer material="1" thickness="123.456" /><layer material="0" thickness="50.0" /><layer material="2" thickness="456.789" /></layers></mc:horizontalLayers>')
self.element2 = etree.parse(source).getroot()
def tearDown(self):
unittest.TestCase.tearDown(self)
def testcan_parse(self):
self.assertTrue(self.h.can_parse(self.element1))
self.assertTrue(self.h.can_parse(self.element2))
def testparse(self):
# Horizontal layers 1
obj = self.h.parse(self.element1)
self.assertTrue(obj.has_substrate())
self.assertEqual('Copper', str(obj.substrate.material))
self.assertEqual(2, len(obj.layers))
self.assertEqual('Zinc', str(obj.layers[0].material))
self.assertAlmostEqual(123.456, obj.layers[0].thickness_m)
self.assertEqual('Gallium', str(obj.layers[1].material))
self.assertAlmostEqual(456.789, obj.layers[1].thickness_m)
self.assertAlmostEqual(1.1, obj.tilt_rad, 4)
self.assertAlmostEqual(2.2, obj.rotation_rad, 4)
# Horizontal layers 2
obj = self.h.parse(self.element2)
self.assertFalse(obj.has_substrate())
self.assertEqual(3, len(obj.layers))
self.assertEqual('Zinc', str(obj.layers[0].material))
self.assertAlmostEqual(123.456, obj.layers[0].thickness_m)
self.assertEqual('Vacuum', str(obj.layers[1].material))
self.assertAlmostEqual(50.0, obj.layers[1].thickness_m)
self.assertEqual('Gallium', str(obj.layers[2].material))
self.assertAlmostEqual(456.789, obj.layers[2].thickness_m)
self.assertAlmostEqual(0.0, obj.tilt_rad, 4)
self.assertAlmostEqual(0.0, obj.rotation_rad, 4)
def testcan_convert(self):
self.assertTrue(self.h.can_convert(self.obj1))
self.assertTrue(self.h.can_convert(self.obj2))
def testconvert(self):
# Horizontal layers 1
element = self.h.convert(self.obj1)
self.assertAlmostEqual(1.1, float(element.get('tilt')), 4)
self.assertAlmostEqual(2.2, float(element.get('rotation')), 4)
self.assertEqual(3, len(list(element.find('materials'))))
self.assertEqual(2, len(list(element.find('layers'))))
# Horizontal layers 2
element = self.h.convert(self.obj2)
self.assertAlmostEqual(0.0, float(element.get('tilt')), 4)
self.assertAlmostEqual(0.0, float(element.get('rotation')), 4)
self.assertEqual(2, len(list(element.find('materials'))))
self.assertEqual(3, len(list(element.find('layers'))))
