mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-22 17:52:07 +08:00
firestar5683
17 lines
594 B
Python
17 lines
594 B
Python
from openpilot.common import file_chunker
|
|
|
|
|
|
def test_chunked_stream_round_trip(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(file_chunker, "CHUNK_SIZE", 7)
|
|
path = tmp_path / "artifact.pkl"
|
|
payload = b"a model artifact spanning several chunks"
|
|
path.write_bytes(payload)
|
|
|
|
targets = file_chunker.get_chunk_targets(path, len(payload))
|
|
file_chunker.chunk_file(path, targets)
|
|
|
|
assert file_chunker.file_chunked_exists(path)
|
|
assert file_chunker.read_file_chunked(path) == payload
|
|
with file_chunker.open_file_chunked(path) as stream:
|
|
assert stream.read(9) + stream.read() == payload
|