63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
|
import unittest
|
||
|
|
from test import test_support
|
||
|
|
|
||
|
|
import io
|
||
|
|
|
||
|
|
class IOTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def write_ops(self, f):
|
||
|
|
f.write(b"blah.")
|
||
|
|
f.seek(0)
|
||
|
|
f.write(b"Hello.")
|
||
|
|
self.assertEqual(f.tell(), 6)
|
||
|
|
f.seek(-1, 1)
|
||
|
|
self.assertEqual(f.tell(), 5)
|
||
|
|
f.write(" world\n\n\n")
|
||
|
|
f.seek(0)
|
||
|
|
f.write("h")
|
||
|
|
f.seek(-2, 2)
|
||
|
|
f.truncate()
|
||
|
|
|
||
|
|
def read_ops(self, f):
|
||
|
|
data = f.read(5)
|
||
|
|
self.assertEqual(data, b"hello")
|
||
|
|
f.readinto(data)
|
||
|
|
self.assertEqual(data, b" worl")
|
||
|
|
f.readinto(data)
|
||
|
|
self.assertEqual(data, b"d\n")
|
||
|
|
f.seek(0)
|
||
|
|
self.assertEqual(f.read(20), b"hello world\n")
|
||
|
|
f.seek(-6, 2)
|
||
|
|
self.assertEqual(f.read(5), b"world")
|
||
|
|
f.seek(-6, 1)
|
||
|
|
self.assertEqual(f.read(5), b" worl")
|
||
|
|
self.assertEqual(f.tell(), 10)
|
||
|
|
|
||
|
|
def test_raw_file_io(self):
|
||
|
|
f = io.open(test_support.TESTFN, "wb", buffering=0)
|
||
|
|
self.assertEqual(f.readable(), False)
|
||
|
|
self.assertEqual(f.writable(), True)
|
||
|
|
self.assertEqual(f.seekable(), True)
|
||
|
|
self.write_ops(f)
|
||
|
|
f.close()
|
||
|
|
f = io.open(test_support.TESTFN, "rb", buffering=0)
|
||
|
|
self.assertEqual(f.readable(), True)
|
||
|
|
self.assertEqual(f.writable(), False)
|
||
|
|
self.assertEqual(f.seekable(), True)
|
||
|
|
self.read_ops(f)
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
def test_raw_bytes_io(self):
|
||
|
|
f = io.BytesIO()
|
||
|
|
self.write_ops(f)
|
||
|
|
data = f.getvalue()
|
||
|
|
self.assertEqual(data, b"hello world\n")
|
||
|
|
f = io.BytesIO(data)
|
||
|
|
self.read_ops(f)
|
||
|
|
|
||
|
|
def test_main():
|
||
|
|
test_support.run_unittest(IOTest)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_main()
|