Fix test_bool and test_bufio.

This commit is contained in:
Guido van Rossum 2007-05-24 16:11:18 +00:00
parent 6f376c4031
commit 305b9246a7
2 changed files with 7 additions and 7 deletions

View File

@ -26,10 +26,10 @@ class BoolTest(unittest.TestCase):
def test_print(self):
try:
fo = open(test_support.TESTFN, "wb")
fo = open(test_support.TESTFN, "w")
print(False, True, file=fo)
fo.close()
fo = open(test_support.TESTFN, "rb")
fo = open(test_support.TESTFN, "r")
self.assertEqual(fo.read(), 'False True\n')
finally:
fo.close()

View File

@ -19,12 +19,12 @@ class BufferSizeTest(unittest.TestCase):
try:
# write once with \n and once without
f.write(s)
f.write("\n")
f.write(b"\n")
f.write(s)
f.close()
f = open(test_support.TESTFN, "rb")
line = f.readline()
self.assertEqual(line, s + "\n")
self.assertEqual(line, s + b"\n")
line = f.readline()
self.assertEqual(line, s)
line = f.readline()
@ -48,16 +48,16 @@ class BufferSizeTest(unittest.TestCase):
teststring = pattern * q + pattern[:r]
self.assertEqual(len(teststring), length)
self.try_one(teststring)
self.try_one(teststring + "x")
self.try_one(teststring + b"x")
self.try_one(teststring[:-1])
def test_primepat(self):
# A pattern with prime length, to avoid simple relationships with
# stdio buffer sizes.
self.drive_one("1234567890\00\01\02\03\04\05\06")
self.drive_one(b"1234567890\00\01\02\03\04\05\06")
def test_nullpat(self):
self.drive_one("\0" * 1000)
self.drive_one(bytes(1000))
def test_main():
test_support.run_unittest(BufferSizeTest)