| import unittest |
| from pathlib import Path |
| import tempfile |
| import numpy as np |
|
|
| from tests.fixtures.tiny_assets import make_tiny_assets |
| from src.maya_fast.assets import default_assets_dir |
| from src.maya_fast.codec import MimiDecoder |
|
|
|
|
| class TestMayaFastCodec(unittest.TestCase): |
| def setUp(self): |
| self.temp_dir = tempfile.TemporaryDirectory(ignore_cleanup_errors=True) |
| self.assets_dir = make_tiny_assets(Path(self.temp_dir.name)) |
|
|
| def tearDown(self): |
| self.temp_dir.cleanup() |
|
|
| def test_codec_length(self): |
| |
| codec_path = self.assets_dir / "mimi_decoder.onnx" |
| mimi = MimiDecoder(codec_path, threads=1) |
|
|
| F = 5 |
| codes = np.random.randint(0, 64, size=(16, F), dtype=np.int64) |
| wav = mimi.decode(codes) |
|
|
| self.assertEqual(wav.shape, (F * 1920,)) |
| self.assertEqual(wav.dtype, np.float32) |
|
|
|
|
| class TestMayaFastCodecReal(unittest.TestCase): |
| def test_real_decoder_length(self): |
| path = default_assets_dir() / "mimi_decoder.onnx" |
| if not path.exists(): |
| self.skipTest(f"real Mimi decoder not found at {path}") |
| mimi = MimiDecoder(path, threads=1) |
| codes = np.random.default_rng(0).integers(0, 2048, size=(16, 5), dtype=np.int64) |
| wav = mimi.decode(codes) |
| self.assertEqual(wav.shape, (5 * 1920,)) |
| self.assertTrue(np.all(np.isfinite(wav))) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
