def setUp(self):
unittest.TestCase.setUp(self)
self.h = SphereXMLHandler()
self.obj = Sphere(Material.pure(29), 123.456, 1.1, 2.2)
etree.register_namespace('mc', 'http://pymontecarlo.sf.net')
source = BytesIO(b'<mc:sphere xmlns:mc="http://pymontecarlo.sf.net" rotation="2.2" tilt="1.1"><materials><mc:material _index="1" density="8960.0" name="Copper"><composition><element weightFraction="1.0" z="29" /></composition></mc:material></materials><body diameter="123.456" material="1" /></mc:sphere>')
self.element = etree.parse(source).getroot()
class TestSphereXMLHandler(unittest.TestCase):
def setUp(self):
unittest.TestCase.setUp(self)
self.h = SphereXMLHandler()
self.obj = Sphere(Material.pure(29), 123.456, 1.1, 2.2)
etree.register_namespace('mc', 'http://pymontecarlo.sf.net')
source = BytesIO(b'<mc:sphere xmlns:mc="http://pymontecarlo.sf.net" rotation="2.2" tilt="1.1"><materials><mc:material _index="1" density="8960.0" name="Copper"><composition><element weightFraction="1.0" z="29" /></composition></mc:material></materials><body diameter="123.456" material="1" /></mc:sphere>')
self.element = etree.parse(source).getroot()
def tearDown(self):
unittest.TestCase.tearDown(self)
def testcan_parse(self):
self.assertTrue(self.h.can_parse(self.element))
def testparse(self):
obj = self.h.parse(self.element)
self.assertEqual('Copper', str(obj.body.material))
self.assertAlmostEqual(123.456, obj.body.diameter_m, 4)
self.assertAlmostEqual(1.1, obj.tilt_rad, 4)
self.assertAlmostEqual(2.2, obj.rotation_rad, 4)
def testcan_convert(self):
self.assertTrue(self.h.can_convert(self.obj))
def testconvert(self):
element = self.h.convert(self.obj)
self.assertAlmostEqual(1.1, float(element.get('tilt')), 4)
self.assertAlmostEqual(2.2, float(element.get('rotation')), 4)
subelement = element.find('body')
self.assertAlmostEqual(123.456, float(subelement.get('diameter')), 4)
