test_codecs now removes the temporay file (created by the test)

This commit is contained in:
Victor Stinner 2011-05-23 14:51:42 +02:00
parent 3c8d12d6b8
commit 2cca057284
1 changed files with 6 additions and 7 deletions

View File

@ -515,13 +515,11 @@ class UTF16Test(ReadTest):
s1 = 'Hello\r\nworld\r\n'
s = s1.encode(self.encoding)
try:
with open(support.TESTFN, 'wb') as fp:
fp.write(s)
with codecs.open(support.TESTFN, 'U', encoding=self.encoding) as reader:
self.assertEqual(reader.read(), s1)
finally:
support.unlink(support.TESTFN)
self.addCleanup(support.unlink, support.TESTFN)
with open(support.TESTFN, 'wb') as fp:
fp.write(s)
with codecs.open(support.TESTFN, 'U', encoding=self.encoding) as reader:
self.assertEqual(reader.read(), s1)
class UTF16LETest(ReadTest):
encoding = "utf-16-le"
@ -1628,6 +1626,7 @@ class BomTest(unittest.TestCase):
"utf-32",
"utf-32-le",
"utf-32-be")
self.addCleanup(support.unlink, support.TESTFN)
for encoding in tests:
# Check if the BOM is written only once
with codecs.open(support.TESTFN, 'w+', encoding=encoding) as f: