diff --git a/xmltok/test_xmltok.py b/xmltok/test_xmltok.py index 05953e90..2f5afe9f 100644 --- a/xmltok/test_xmltok.py +++ b/xmltok/test_xmltok.py @@ -15,7 +15,11 @@ expected = [ ('END_TAG', ('s', 'Envelope')), ] +dir = "." +if "/" in __file__: + dir = __file__.rsplit("/", 1)[0] + ex = iter(expected) -for i in xmltok.tokenize(open("test.xml")): +for i in xmltok.tokenize(open(dir + "/test.xml")): #print(i) assert i == next(ex)