diff options
Diffstat (limited to 'testers/factorytester.py')
| -rw-r--r-- | testers/factorytester.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/testers/factorytester.py b/testers/factorytester.py index b6029ca..88e660d 100644 --- a/testers/factorytester.py +++ b/testers/factorytester.py @@ -3,17 +3,19 @@ import parsers import factories import xml.etree.cElementTree as ET -xmlpath = 'C:/Users/Ilsar/Documents/datahack/xml_example.xml' +xmlpath = r"C:\roy\NLI-nnl10 - 1MB.xml" +whitelist = ['100', '374', '400', '151', '451', '550', '551', '678'] + xmltree = ET.parse(xmlpath) entities = list() inl_factory = factories.INLFactory() for record in xmltree.getroot(): - inl_parser = parsers.INLXmlParser(record) + inl_parser = parsers.INLXmlParser(record, whitelist) clean_record = inl_parser.clearxml() entities.append(inl_factory.get_entity(clean_record)) for entity in entities: - entity. print_entity() + print(entity) |