| import pytest | |
| from dlb.io import atomic_json_write, sha256_file | |
| from dlb.schema import SampleRecord | |
| def test_sample_record_rejects_empty_text(): | |
| with pytest.raises(ValueError): | |
| SampleRecord( | |
| sample_id=0, | |
| text="", | |
| token_ids=[1], | |
| seed=42, | |
| generation_seconds=0.1, | |
| ) | |
| def test_sample_record_rejects_negative_generation_time(): | |
| with pytest.raises(ValueError): | |
| SampleRecord( | |
| sample_id=0, | |
| text="generated text", | |
| token_ids=[1], | |
| seed=42, | |
| generation_seconds=-0.1, | |
| ) | |
| def test_atomic_json_write_replaces_file_with_deterministic_json(tmp_path): | |
| target = tmp_path / "nested" / "record.json" | |
| atomic_json_write(target, {"z": 1, "a": [2]}) | |
| assert target.read_text() == '{\n "a": [\n 2\n ],\n "z": 1\n}\n' | |
| def test_sha256_file_streams_known_content_digest(tmp_path): | |
| target = tmp_path / "asset.bin" | |
| target.write_bytes(b"abc") | |
| assert sha256_file(target) == ( | |
| "ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad" | |
| ) | |