| | """Tests for configuration module.""" |
| |
|
| |
|
| | class TestTranslatorConfig: |
| | """Tests for TranslatorConfig.""" |
| |
|
| | def test_default_config(self): |
| | """Test default configuration values.""" |
| | from zen_translator.config import TranslatorConfig |
| |
|
| | config = TranslatorConfig() |
| |
|
| | assert config.target_language == "en" |
| | assert config.device == "cuda" |
| | assert config.dtype == "bfloat16" |
| | assert config.enable_lip_sync is True |
| | assert config.voice_reference_seconds == 3.0 |
| |
|
| | def test_config_from_env(self, monkeypatch): |
| | """Test configuration from environment variables.""" |
| | from zen_translator.config import TranslatorConfig |
| |
|
| | monkeypatch.setenv("ZEN_TRANSLATOR_TARGET_LANGUAGE", "es") |
| | monkeypatch.setenv("ZEN_TRANSLATOR_DEVICE", "cpu") |
| |
|
| | config = TranslatorConfig() |
| |
|
| | assert config.target_language == "es" |
| | assert config.device == "cpu" |
| |
|
| | def test_supported_languages(self): |
| | """Test supported language lists.""" |
| | from zen_translator.config import TranslatorConfig |
| |
|
| | config = TranslatorConfig() |
| |
|
| | |
| | assert "en" in config.supported_input_languages |
| | assert "zh" in config.supported_input_languages |
| | assert "ja" in config.supported_input_languages |
| | assert "yue" in config.supported_input_languages |
| |
|
| | |
| | assert "en" in config.supported_output_languages |
| | assert "zh" in config.supported_output_languages |
| | assert len(config.supported_output_languages) == 10 |
| |
|
| | def test_lip_sync_quality_options(self): |
| | """Test lip sync quality options.""" |
| | from zen_translator.config import TranslatorConfig |
| |
|
| | for quality in ["fast", "balanced", "quality"]: |
| | config = TranslatorConfig(lip_sync_quality=quality) |
| | assert config.lip_sync_quality == quality |
| |
|
| |
|
| | class TestNewsAnchorConfig: |
| | """Tests for NewsAnchorConfig.""" |
| |
|
| | def test_default_config(self): |
| | """Test default news anchor config.""" |
| | from zen_translator.config import NewsAnchorConfig |
| |
|
| | config = NewsAnchorConfig() |
| |
|
| | assert config.min_clip_duration == 5.0 |
| | assert config.max_clip_duration == 30.0 |
| | assert len(config.target_anchors) > 0 |
| |
|
| | def test_training_settings(self): |
| | """Test training hyperparameters.""" |
| | from zen_translator.config import NewsAnchorConfig |
| |
|
| | config = NewsAnchorConfig() |
| |
|
| | assert config.batch_size == 4 |
| | assert config.learning_rate == 2e-5 |
| | assert config.num_epochs == 3 |
| |
|
