def get_xml(workout): p = swim_parser.parser res = p.parse(workout) sv = SwimXmlVisitor() cv = SwimXmlCloseVisitor() out = [] traversal.xml_traverse(res, sv, cv, out) out.insert(0, '<?xml version="1.0" encoding="utf-8"?>') return '\n'.join(out)
def test_set_with_only_distance(self): """ Takes a set that is defined only by yards, e.g. 1000. """ expected = '''<setlist xmlns="http://swimparser.appspot.com/xml"> <set> <count> <reps>1</reps> <distance>1000</distance> </count> <stroke></stroke> <zone></zone> <time>0</time> </set> </setlist>''' s = '1000' out = [] result = self.parser.parse(s) traversal.xml_traverse(result, self.xml_visitor, self.xml_close_visitor, out) actual = '\n'.join(out) self.assertEqual(actual, expected)