| """Release integrity, local loading, metadata exclusions and wheel checks.""" |
| import importlib.util |
| import json |
| from pathlib import Path |
| import shutil |
| import zipfile |
| import numpy as np |
| import pytest |
| from timestral_mlx import TimestralForecaster |
|
|
| ROOT=Path(__file__).resolve().parents[1] |
|
|
|
|
| def test_local_bundle_roundtrip_and_corruption(tmp_path): |
| source=ROOT/'models'/'5M' |
| if not source.exists(): pytest.skip('convert 5M first') |
| for name in ('config.json','model.safetensors','LICENSE','manifest.json'): |
| shutil.copyfile(source/name,tmp_path/name) |
| fc=TimestralForecaster.from_pretrained(tmp_path) |
| assert fc.forecast(np.arange(65,dtype=np.float32),9).quantiles.shape==(1,1,9,21) |
| with (tmp_path/'model.safetensors').open('r+b') as f: |
| f.seek(-1,2); byte=f.read(1); f.seek(-1,2); f.write(bytes([byte[0]^1])) |
| with pytest.raises(ValueError,match='integrity'): |
| TimestralForecaster.from_pretrained(tmp_path) |
|
|
|
|
| def test_variant_mismatch(): |
| source=ROOT/'models'/'5M' |
| if not source.exists(): pytest.skip('convert 5M first') |
| with pytest.raises(ValueError,match='variant'): |
| TimestralForecaster.from_pretrained(source.parent,variant='not-a-model') |
|
|
|
|
| def test_apple_metadata_exclusions(): |
| spec=importlib.util.spec_from_file_location('package_release',ROOT/'scripts'/'package_release.py') |
| module=importlib.util.module_from_spec(spec);spec.loader.exec_module(module) |
| for name in ('._README.md','5M/._model.safetensors','__MACOSX/foo','x/.DS_Store','x/a.pyc','x/a.tmp','x/.AppleDouble/README'): |
| assert module.excluded(name),name |
| for name in ('5M/model.safetensors','timestral_mlx/model.py','LICENSE-WEIGHTS'): |
| assert not module.excluded(name),name |
|
|
|
|
| def test_installable_wheel_inventory(): |
| wheels=[p for p in (ROOT/'dist').glob('*.whl') if not p.name.startswith('._')] |
| if not wheels: pytest.skip('build wheel first') |
| with zipfile.ZipFile(wheels[0]) as z: |
| assert not any(any(p.startswith('._') or p in {'.DS_Store','__MACOSX'} for p in n.split('/')) for n in z.namelist()) |
| for path in (ROOT/'timestral_mlx').glob('*.py'): |
| if not path.name.startswith('._'): |
| assert z.read('timestral_mlx/'+path.name)==path.read_bytes() |
| assert z.read('timestral_mlx-0.1.0.dist-info/licenses/LICENSE-WEIGHTS')==(ROOT/'LICENSE-WEIGHTS').read_bytes() |
|
|
