Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
55 lines
2.5 KiB
Python
55 lines
2.5 KiB
Python
"""Asset-reader regression tests; standard library only, no model weights."""
|
|
import importlib.util
|
|
import json
|
|
from pathlib import Path
|
|
import struct
|
|
import tempfile
|
|
from types import SimpleNamespace
|
|
import unittest
|
|
import zipfile
|
|
|
|
MODULE_PATH = Path(__file__).resolve().parents[2] / 'tools/community_models/prepare_kitten_tts2_gguf.py'
|
|
SPEC = importlib.util.spec_from_file_location('kitten_prepare', MODULE_PATH)
|
|
PREPARE = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(PREPARE)
|
|
|
|
|
|
class VoiceArrayTests(unittest.TestCase):
|
|
def read(self, descriptor, shape, payload, dtype, version=1):
|
|
header = repr({'descr': descriptor, 'fortran_order': False, 'shape': shape}).encode('ascii')
|
|
width = 2 if version == 1 else 4
|
|
data = b'\x93NUMPY' + bytes([version, 0]) + len(header).to_bytes(width, 'little') + header + payload
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = Path(directory) / 'voice.npz'
|
|
with zipfile.ZipFile(path, 'w') as archive:
|
|
archive.writestr('value.npy', data)
|
|
return PREPARE.read_voice_array(path, 'value', dtype)
|
|
|
|
def test_upstream_numeric_types_and_endianness(self):
|
|
for version in (1, 2, 3):
|
|
self.assertEqual(self.read('<f4', (2,), struct.pack('<2f', .5, -1), 'f4', version), [.5, -1])
|
|
self.assertEqual(self.read('>i4', (2,), struct.pack('>2i', 4299, 6560), 'i4', version), [4299, 6560])
|
|
|
|
def test_rejects_object_or_wrong_shape(self):
|
|
for descriptor, shape in (('|O', (1,)), ('<f4', (1, 1)), ('<i4', (1,))):
|
|
with self.assertRaises(ValueError):
|
|
self.read(descriptor, shape, b'\0' * 4, 'f4')
|
|
|
|
def test_rejects_truncated_and_nonfinite_values(self):
|
|
with self.assertRaises(ValueError):
|
|
self.read('<f4', (2,), struct.pack('<f', 1), 'f4')
|
|
with self.assertRaises(ValueError):
|
|
self.read('<f4', (1,), struct.pack('<f', float('nan')), 'f4')
|
|
|
|
def test_incomplete_prepared_index_is_rejected(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
voices = root / 'voices.json'
|
|
voices.write_text(json.dumps({'Bruno': {}}))
|
|
args = SimpleNamespace(prepared_voices=voices, voices_output=None)
|
|
with self.assertRaisesRegex(ValueError, 'German'):
|
|
PREPARE.prepare_voices(args, root, root / 'unused.safetensors', root, {'Bruno', 'German'})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|