def parse(self, element): transitions = [] for subelement in element.iter('transition'): z = int(self._parse_numerical_parameter(subelement, 'z')) src = int(self._parse_numerical_parameter(subelement, 'src')) dest = int(self._parse_numerical_parameter(subelement, 'dest')) transitions.append(Transition(z, src, dest)) return _TransitionsDetector(transitions)
def setUp(self): unittest.TestCase.setUp(self) self.h = _TransitionsDetectorXMLHandler() self.t1 = Transition(24, siegbahn="Ka1") self.obj = _TransitionsDetector([self.t1]) etree.register_namespace("mc", "http://pymontecarlo.sf.net") source = BytesIO( b'<mc:_transitionsDetector xmlns:mc="http://pymontecarlo.sf.net"><transition z="24" src="4" dest="1" /></mc:_transitionsDetector>' ) self.element = etree.parse(source).getroot()
def test__init__(self): d = _TransitionsDetector() self.assertEqual(0, len(d.transitions))
def setUp(self): TestCase.setUp(self) self.t1 = Transition(24, siegbahn='Ka1') self.t2 = Transition(24, siegbahn='La1') self.d = _TransitionsDetector([self.t1, self.t2])