Fix ResourceWarnings in test_sax

2adb6fe2 · Antoine Pitrou · bf4cba7e · 2adb6fe2
Commit 2adb6fe2 authored May 13, 2013 by Antoine Pitrou
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

Lib/test/test_sax.py Lib/test/test_sax.py +4 -1

No files found.
--- a/Lib/test/test_sax.py
+++ b/Lib/test/test_sax.py
@@ -574,7 +574,10 @@ class StreamReaderWriterXmlgenTest(XmlgenTest, unittest.TestCase):
    def ioclass(self):
        writer = codecs.open(self.fname, 'w', encoding='ascii',
                             errors='xmlcharrefreplace', buffering=0)
-        self.addCleanup(support.unlink, self.fname)
+        def cleanup():
+            writer.close()
+            support.unlink(self.fname)
+        self.addCleanup(cleanup)
        writer.getvalue = self.getvalue
        return writer