def setUp(self): unittest.TestCase.setUp(self) self.h = _ModelXMLHandler() self.obj1 = _Model(MTYPE, 'name1', 'ref1') self.obj2 = _Model(MTYPE, 'name2') etree.register_namespace('mc', 'http://pymontecarlo.sf.net') source = BytesIO(b'<mc:_model xmlns:mc="http://pymontecarlo.sf.net" name="name1" type="type1" reference="ref1" />') self.element1 = etree.parse(source).getroot() source = BytesIO(b'<mc:_model xmlns:mc="http://pymontecarlo.sf.net" name="name2" type="type1" />') self.element2 = etree.parse(source).getroot()
def parse(self, element): name = element.get('name') type_ = ModelType(element.get('type')) reference = element.get('reference', '') return _Model(type_, name, reference)