30 lines
755 B
Python
30 lines
755 B
Python
import os
|
|
|
|
from faster_whisper import available_models, download_model
|
|
|
|
|
|
def test_available_models():
|
|
models = available_models()
|
|
assert isinstance(models, list)
|
|
assert "tiny" in models
|
|
|
|
|
|
def test_download_model(tmpdir):
|
|
output_dir = str(tmpdir.join("model"))
|
|
|
|
model_dir = download_model("tiny", output_dir=output_dir)
|
|
|
|
assert model_dir == output_dir
|
|
assert os.path.isdir(model_dir)
|
|
assert not os.path.islink(model_dir)
|
|
|
|
for filename in os.listdir(model_dir):
|
|
path = os.path.join(model_dir, filename)
|
|
assert not os.path.islink(path)
|
|
|
|
|
|
def test_download_model_in_cache(tmpdir):
|
|
cache_dir = str(tmpdir.join("model"))
|
|
download_model("tiny", cache_dir=cache_dir)
|
|
assert os.path.isdir(cache_dir)
|