--- tests/testxmlgen.py +++ tests/testxmlgen.py @@ -20,6 +20,7 @@ import xml.sax, xml.sax.saxutils import StringIO +import sys import unittest class MyGen(xml.sax.saxutils.XMLGenerator): @@ -70,14 +71,16 @@ <a:greetings xmlns:a="http://example.com/ns"> <a:greet xml:lang="en">Hello world</a:greet> </a:greetings>""" - parser.feed(testcontent) - parser.close() - expectedresult="""<?xml version="1.0" encoding="utf-8"?> + if sys.version_info < (2, 7): + self.assertRaises(KeyError, parser.feed, testcontent) + else: + parser.feed(testcontent) + parser.close() + expectedresult="""<?xml version="1.0" encoding="utf-8"?> <a:greetings xmlns:a="http://example.com/ns"> <a:greet xml:lang="en">Hello world</a:greet> </a:greetings>""" - self.assertEqual( outfp.getvalue(), expectedresult) -# self.assertRaises(KeyError, parser.feed, testcontent) + self.assertEqual( outfp.getvalue(), expectedresult) def test_myxml(self): """ Test that my patch works """