def test_chunkedfile_errors():
# read on writable file
writer = chunky.open("tmp/test_error_{0}.txt", "w")
with pytest.raises(IOError):
writer.readline()
writer.close()
# write on readable file
reader = chunky.open("tmp/test_error_{0}.txt", "r")
with pytest.raises(IOError):
reader.write("Test")
reader.close()
def test_chunkwriter_onefile():
writer = chunky.open("tmp/test_{0}.txt", 'w', 10)
assert isinstance(writer, chunky.ChunkedTextFile)
for i in range(0, 10):
writer.write(str(i) + os.linesep)
writer.close()
assert os.path.exists("tmp/test_0.txt")
expected_content = """
0
1
2
3
4
5
6
7
8
9
"""
result_content = file_get_contents("tmp/test_0.txt")
assert expected_content.strip() == result_content.strip()
def test_chunkwriter_callback_error():
writer = chunky.open("tmp/test_{0}.txt", 'w', cb_chunk_closed="NotACallback")
for i in range(0, 5):
writer.write(str(i) + os.linesep)
with pytest.raises(ValueError):
writer.close()
def test_chunky_open():
# create dummy file
fp = open("tmp/test_0.txt", 'w')
fp.close()
reader = chunky.open("tmp/test_{0}.txt", 'r')
assert isinstance(reader, chunky.ChunkedTextFile)
assert reader.readable()
assert not reader.writable()
assert not reader.seekable()
reader.close()
writer = chunky.open("tmp/test_{0}.txt", 'w')
assert isinstance(writer, chunky.ChunkedTextFile)
assert not writer.readable()
assert writer.writable()
assert not writer.seekable()
writer.close()
# test unsupported mode
with pytest.raises(ValueError):
chunky.open("tmp/test_{0}.txt", 'a')
def test_chunkreader_onefile():
# create a dummy file
fp = open("tmp/test_reader_onefile_0.csv", 'w')
for i in range(0, 10):
fp.write(str(i) + os.linesep)
fp.close()
content = []
reader = chunky.open("tmp/test_reader_onefile_{0}.csv", 'r')
for line in reader:
content.append(line.strip())
expected_content = [str(i) for i in range(0, 10)]
assert expected_content == content
def test_with_csv_dictwriter():
import csv
csvfile = chunky.open("tmp/csv_test_{0}.csv", mode='w', chunk_size=6)
writer = csv.DictWriter(csvfile, fieldnames=["id", "name"])
def chunk_start(filename):
global chunk_start_called
chunk_start_called = True
writer.writeheader()
assert filename is not None
def chunk_written(filename, num_lines_written):
global chunk_closed_called
chunk_closed_called = True
assert filename is not None
assert num_lines_written > 0
csvfile.cb_chunk_start = chunk_start
csvfile.cb_chunk_closed = chunk_written
writer.writeheader()
for i in range(0, 16):
writer.writerow({"id": i, "name": "Test"})
csvfile.close()
assert chunk_closed_called
assert chunk_start_called
assert os.path.exists("tmp/csv_test_0.csv")
assert os.path.exists("tmp/csv_test_1.csv")
assert os.path.exists("tmp/csv_test_2.csv")
num_lines = []
for i in range(0, 3):
fp = open("tmp/csv_test_0.csv")
reader = csv.DictReader(fp)
assert reader.fieldnames == ["id", "name"]
count = 0
for row in reader:
assert row["id"]
assert row["name"]
count += 1
num_lines.append(count)
assert num_lines == [5, 5, 5]
def test_chunkwriter_pattern():
wrong_pattern = "tmp/{0}/test.txt"
with pytest.raises(ValueError):
chunky.open(wrong_pattern, mode='w')
wrong_pattern = "tmp/test.txt"
with pytest.raises(ValueError):
chunky.open(wrong_pattern, mode='w')
wrong_pattern = "tmp/test_{0}_{1}.txt"
with pytest.raises(ValueError):
chunky.open(wrong_pattern, mode='w')
def test_chunkreader_multiple_files():
fp = open("tmp/test_reader_0.csv", 'w')
for i in range(0, 10):
fp.write(str(i) + os.linesep)
fp.close()
fp = open("tmp/test_reader_1.csv", 'w')
for i in range(10, 15):
fp.write(str(i) + os.linesep)
fp.close()
fp = open("tmp/test_reader_2.csv", 'w')
for i in range(15, 20):
fp.write(str(i) + os.linesep)
fp.close()
content = []
reader = chunky.open("tmp/test_reader_{0}.csv", 'r')
for line in reader:
content.append(line.strip())
expected_content = [str(i) for i in range(0, 20)]
assert expected_content == content
def test_chunkwriter_callback():
def callback(filename, num_lines_written):
global callback_called, callback_count
callback_called = True
callback_count += 1
assert os.path.exists(filename)
assert num_lines_written > 0
assert len(file_get_contents(filename)) > 0
writer = chunky.open(
"tmp/test_{0}.txt", 'w', chunk_size=10,
cb_chunk_closed=callback)
for i in range(0, 35):
writer.write(str(i) + os.linesep)
writer.close()
assert callback_called
assert os.path.exists("tmp/test_0.txt")
assert os.path.exists("tmp/test_1.txt")
assert os.path.exists("tmp/test_2.txt")
assert os.path.exists("tmp/test_3.txt")
assert callback_count == 4
def test_chunkwriter_exception_on_closed_file():
writer = chunky.open("tmp/test_closed_file_{0}.txt", 'w')
writer.close()
with pytest.raises(ValueError):
writer.write("test" + os.linesep)
