Adam Ladachowski
31 lines
702 B
Python
31 lines
702 B
Python
"""Pytest configuration and fixtures."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import struct
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_safetensor(tmp_path):
|
|
"""Create a minimal valid safetensor file for testing."""
|
|
|
|
# Create minimal safetensor with empty tensors and some metadata
|
|
header = {
|
|
"__metadata__": {
|
|
"format": "pt",
|
|
"test_key": "test_value",
|
|
}
|
|
}
|
|
header_bytes = json.dumps(header).encode("utf-8")
|
|
header_size = len(header_bytes)
|
|
|
|
file_path = tmp_path / "test_model.safetensors"
|
|
with file_path.open("wb") as f:
|
|
f.write(struct.pack("<Q", header_size))
|
|
f.write(header_bytes)
|
|
|
|
return file_path
|