pocketpy/tests/70_file.py
2024-01-20 22:39:32 +08:00

42 lines
806 B
Python

try:
import os
import io
print("[`os` Test Enabled]")
except ImportError:
exit(0)
a = open('123.txt', 'wt')
a.write('123')
a.write('456')
a.close()
with open('123.txt', 'rt') as f:
assert f.read() == '123456'
with open('123.txt', 'a') as f:
f.write('测试')
# default mode is 'r'
with open('123.txt') as f:
assert f.read() == '123456' + '测试'
assert os.path.exists('123.txt')
os.remove('123.txt')
assert not os.path.exists('123.txt')
with open('123.bin', 'wb') as f:
f.write('123'.encode())
f.write('测试'.encode())
def f():
with open('123.bin', 'rb') as f:
b = f.read()
assert isinstance(b, bytes)
assert b == '123测试'.encode()
f()
assert os.path.exists('123.bin')
os.remove('123.bin')
assert not os.path.exists('123.bin')