You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
765 B
Python
26 lines
765 B
Python
2 years ago
|
import os
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from pydantic.types import SecretStr
|
||
|
|
|
||
|
|
from local_whisper.settings import WhisperSettings
|
||
|
|
|
||
|
|
SETTING_DEFAULTS = {
|
||
|
|
"BASE_ASR_MODEL": "medium.en",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def test_setting_defaults():
|
||
|
|
"""Regression test for settings schema."""
|
||
|
|
with patch.dict(os.environ, {}, clear=True):
|
||
|
|
savant_settings = WhisperSettings()
|
||
|
|
assert len(savant_settings.dict()) == len(SETTING_DEFAULTS)
|
||
|
|
for k, v in SETTING_DEFAULTS.items():
|
||
|
|
_setting_value = getattr(savant_settings, k.lower())
|
||
|
|
unmasked_setting = (
|
||
|
|
_setting_value.get_secret_value()
|
||
|
|
if isinstance(_setting_value, SecretStr)
|
||
|
|
else _setting_value
|
||
|
|
)
|
||
|
|
unmasked_setting == v
|