20 lines
504 B
Python
20 lines
504 B
Python
import tempfile
|
|
from pathlib import Path
|
|
|
|
import layercake.jsonl as jsonl
|
|
|
|
|
|
def test_readlines():
|
|
with tempfile.NamedTemporaryFile() as fp:
|
|
fp.writelines([b'{}\n' for _ in range(4)])
|
|
fp.seek(0)
|
|
|
|
with jsonl.readlines(fp.name) as lines:
|
|
assert sum(1 for _ in lines) == 4
|
|
|
|
with jsonl.readlines(Path('notfound.jsonl')) as lines:
|
|
assert sum(1 for _ in lines) == 0
|
|
|
|
with jsonl.readlines(Path('notfound.jsonl')) as lines:
|
|
assert list(lines) == []
|