mirror of
https://github.com/kuhyx/testsAndMisc-archive.git
synced 2026-07-04 15:03:04 +02:00
Krzysztof kuhy Rudnicki
996617d4a0
- Add comprehensive tests for all packages (3572 tests, 100% branch coverage) - Split oversized test files to stay under 500-line limit - Add per-file ruff ignores for test-appropriate suppressions - Fix _cache_decks.py to properly convert JSON lists to tuples - Add session-scoped conftest fixture for logging handler cleanup (Python 3.14) - Update ruff pre-commit hook to v0.15.2 - Add codespell ignore words for test data - Add generated output files to .gitignore
246 lines
8.8 KiB
Python
246 lines
8.8 KiB
Python
"""Tests for python_pkg.music_gen.music_generator module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Any
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from python_pkg.music_gen.music_generator import (
|
|
check_dependencies,
|
|
interactive_mode,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
import pytest
|
|
|
|
|
|
class TestCheckDependencies:
|
|
"""Tests for check_dependencies()."""
|
|
|
|
def test_all_present(self) -> None:
|
|
with patch("importlib.util.find_spec", return_value=MagicMock()):
|
|
assert check_dependencies() is True
|
|
|
|
def test_torch_missing(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
def mock_find_spec(name: str) -> Any:
|
|
if name == "torch":
|
|
return None
|
|
return MagicMock()
|
|
|
|
with patch("importlib.util.find_spec", side_effect=mock_find_spec):
|
|
assert check_dependencies() is False
|
|
|
|
captured = capsys.readouterr()
|
|
assert "torch" in captured.out
|
|
|
|
def test_transformers_missing(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
def mock_find_spec(name: str) -> Any:
|
|
if name == "transformers":
|
|
return None
|
|
return MagicMock()
|
|
|
|
with patch("importlib.util.find_spec", side_effect=mock_find_spec):
|
|
assert check_dependencies() is False
|
|
|
|
captured = capsys.readouterr()
|
|
assert "transformers" in captured.out
|
|
|
|
def test_scipy_missing(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
def mock_find_spec(name: str) -> Any:
|
|
if name == "scipy":
|
|
return None
|
|
return MagicMock()
|
|
|
|
with patch("importlib.util.find_spec", side_effect=mock_find_spec):
|
|
assert check_dependencies() is False
|
|
|
|
captured = capsys.readouterr()
|
|
assert "scipy" in captured.out
|
|
|
|
def test_bark_missing_with_include_bark(
|
|
self,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
def mock_find_spec(name: str) -> Any:
|
|
if name == "bark":
|
|
return None
|
|
return MagicMock()
|
|
|
|
with patch("importlib.util.find_spec", side_effect=mock_find_spec):
|
|
assert check_dependencies(include_bark=True) is False
|
|
|
|
captured = capsys.readouterr()
|
|
assert "bark" in captured.out.lower()
|
|
|
|
def test_bark_not_checked_without_flag(self) -> None:
|
|
with patch("importlib.util.find_spec", return_value=MagicMock()):
|
|
assert check_dependencies(include_bark=False) is True
|
|
|
|
def test_all_present_with_bark(self) -> None:
|
|
with patch("importlib.util.find_spec", return_value=MagicMock()):
|
|
assert check_dependencies(include_bark=True) is True
|
|
|
|
|
|
class TestInteractiveMode:
|
|
"""Tests for interactive_mode()."""
|
|
|
|
def test_quit_command(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", return_value=":q"):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Exiting" in captured.out
|
|
|
|
def test_quit_word(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", return_value="quit"):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Exiting" in captured.out
|
|
|
|
def test_exit_word(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", return_value="exit"):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
def test_help_command(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=[":h", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Example prompts" in captured.out
|
|
|
|
def test_help_word(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=["help", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Example prompts" in captured.out
|
|
|
|
def test_set_duration(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=[":d 15", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Duration set to 15s" in captured.out
|
|
|
|
def test_set_duration_clamped(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=[":d 100", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Duration set to 30s" in captured.out
|
|
|
|
def test_set_duration_invalid(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=[":d abc", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Invalid duration" in captured.out
|
|
|
|
def test_empty_prompt(self) -> None:
|
|
with patch("builtins.input", side_effect=["", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
def test_number_prompt_valid(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with (
|
|
patch("builtins.input", side_effect=["1", ":q"]),
|
|
patch(
|
|
"python_pkg.music_gen.music_generator.generate_music",
|
|
) as mock_gen,
|
|
):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
mock_gen.assert_called_once()
|
|
|
|
def test_number_prompt_invalid(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=["99", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Invalid number" in captured.out
|
|
|
|
def test_normal_prompt(self) -> None:
|
|
with (
|
|
patch("builtins.input", side_effect=["jazz music", ":q"]),
|
|
patch(
|
|
"python_pkg.music_gen.music_generator.generate_music",
|
|
) as mock_gen,
|
|
):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
mock_gen.assert_called_once()
|
|
|
|
def test_generation_error(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with (
|
|
patch("builtins.input", side_effect=["jazz music", ":q"]),
|
|
patch(
|
|
"python_pkg.music_gen.music_generator.generate_music",
|
|
side_effect=RuntimeError("CUDA OOM"),
|
|
),
|
|
):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Error generating music" in captured.out
|
|
|
|
def test_eof_error(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=EOFError):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Exiting" in captured.out
|
|
|
|
def test_keyboard_interrupt(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=KeyboardInterrupt):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Exiting" in captured.out
|
|
|
|
def test_quit_long(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", return_value=":quit"):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Exiting" in captured.out
|
|
|
|
def test_help_long(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=[":help", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Example prompts" in captured.out
|
|
|
|
def test_duration_clamp_minimum(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with patch("builtins.input", side_effect=[":d 0", ":q"]):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Duration set to 1s" in captured.out
|
|
|
|
def test_generation_value_error(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with (
|
|
patch("builtins.input", side_effect=["jazz", ":q"]),
|
|
patch(
|
|
"python_pkg.music_gen.music_generator.generate_music",
|
|
side_effect=ValueError("bad value"),
|
|
),
|
|
):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Error generating music" in captured.out
|
|
|
|
def test_generation_os_error(self, capsys: pytest.CaptureFixture[str]) -> None:
|
|
with (
|
|
patch("builtins.input", side_effect=["jazz", ":q"]),
|
|
patch(
|
|
"python_pkg.music_gen.music_generator.generate_music",
|
|
side_effect=OSError("disk full"),
|
|
),
|
|
):
|
|
interactive_mode(MagicMock(), MagicMock())
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Error generating music" in captured.out
|