diff options
Diffstat (limited to 'test_llamachat.py')
| -rwxr-xr-x | test_llamachat.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/test_llamachat.py b/test_llamachat.py index bf43c77..f665ee3 100755 --- a/test_llamachat.py +++ b/test_llamachat.py @@ -251,6 +251,11 @@ def test_classify(): assert backend.classify(Path("a.so")) == "unknown" assert backend.classify(Path("a.wav")) == "audio" assert backend.classify(Path("a.WAV")) == "audio" + # Only WAV is accepted; other audio containers must not be labelled + # audio, since the request hardcodes format "wav". + assert backend.classify(Path("a.mp3")) == "unknown" + assert backend.classify(Path("a.ogg")) == "unknown" + assert backend.classify(Path("a.flac")) == "unknown" print("ok file classification") |