Exemplo n.º 1
0
    def setUp(self):
        unittest.TestCase.setUp(self)

        self.h = TransmittedElectronEnergyDetectorXMLHandler()

        self.obj = TransmittedElectronEnergyDetector(1000, (12.34, 56.78))

        etree.register_namespace("mc", "http://pymontecarlo.sf.net")
        source = BytesIO(
            b'<mc:transmittedElectronEnergyDetector xmlns:mc="http://pymontecarlo.sf.net"><channels>1000</channels><limits lower="12.34" upper="56.78" /></mc:transmittedElectronEnergyDetector>'
        )
        self.element = etree.parse(source).getroot()
Exemplo n.º 2
0
class TestTransmittedElectronEnergyDetectorXMLHandler(unittest.TestCase):
    def setUp(self):
        unittest.TestCase.setUp(self)

        self.h = TransmittedElectronEnergyDetectorXMLHandler()

        self.obj = TransmittedElectronEnergyDetector(1000, (12.34, 56.78))

        etree.register_namespace("mc", "http://pymontecarlo.sf.net")
        source = BytesIO(
            b'<mc:transmittedElectronEnergyDetector xmlns:mc="http://pymontecarlo.sf.net"><channels>1000</channels><limits lower="12.34" upper="56.78" /></mc:transmittedElectronEnergyDetector>'
        )
        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.assertAlmostEqual(12.34, obj.limits_eV[0], 4)
        self.assertAlmostEqual(56.78, obj.limits_eV[1], 4)
        self.assertEqual(1000, obj.channels)

    def testcan_convert(self):
        self.assertTrue(self.h.can_convert(self.obj))

    def testconvert(self):
        element = self.h.convert(self.obj)

        subelement = element.find("limits")
        self.assertAlmostEqual(12.34, float(subelement.get("lower")), 4)
        self.assertAlmostEqual(56.78, float(subelement.get("upper")), 4)

        subelement = element.find("channels")
        self.assertEqual(1000, int(subelement.text))