CalvinBackup/OBLITERATUS
1
0
Fork 0
mirror of https://github.com/elder-plinius/OBLITERATUS.git synced 2026-08-30 06:30:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

test cli dispatch

Browse Source
This commit is contained in:
faber
2026-08-19 21:38:19 -04:00
parent c0b3ad9007
commit aa3095747c
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_restore_multimodal.py
+20
View File
@@ -192,3 +192,23 @@ class TestMain:
sys.argv = old_argv
assert (out_dir / "model.safetensors.index.json").exists()
def test_cli_dispatch(self, tmp_path):
"""Test that 'obliteratus restore-multimodal' dispatches correctly."""
from obliteratus.cli import main as cli_main
abl_dir = tmp_path / "abl"
abl_dir.mkdir()
stock_dir = tmp_path / "stock"
stock_dir.mkdir()
out_dir = tmp_path / "out"
_make_fake_model(abl_dir, "model.", n_layers=1)
_make_fake_model(stock_dir, "model.language_model.", n_layers=1)
cli_main(["restore-multimodal",
"--abliterated", str(abl_dir),
"--stock", str(stock_dir),
"--output", str(out_dir)])
assert (out_dir / "model.safetensors.index.json").exists()