"""cli/visualize.py の仕様ベーステスト． 仕様出典: - Phase 1 実装サマリー（sm-visualize CLI 仕様） - docs/04_SPEC/SPEC_01_アーキテクチャ設計.md（CLI 層の責務） - docs/01_GUIDE/GUIDE_08_テスト方針.md（テスト方針） テスト方針: - main() を直接呼び出す（subprocess 不使用：速度・カバレッジのため） - ファイル I/O は tmp_path で実ファイルを使う（GUIDE_08） - PyVista は off_screen=True 相当（screenshot 指定 or --off-screen）で実行 CLI 仕様（sm-visualize）: --scene / -s 必須 --no-bake ベイクをスキップ --screenshot PNG を保存（off_screen 強制） --output / -o ベイク結果を npz として保存 --off-screen ウィンドウを開かない """ from __future__ import annotations from pathlib import Path import numpy as np import pytest from sm_tias_lightsim.cli.visualize import main # --------------------------------------------------------------------------- # テスト用定数 # --------------------------------------------------------------------------- MINIMAL_YAML = Path(__file__).parent.parent / "fixtures" / "minimal.yaml" # --------------------------------------------------------------------------- # 1. 必須引数の欠落（軽量 — argparse 動作） # --------------------------------------------------------------------------- def test_missing_scene_arg_exits_with_error(): """--scene を渡さないと SystemExit が発生すること． 仕様: --scene は必須引数（required=True） """ with pytest.raises(SystemExit) as exc_info: main([]) assert exc_info.value.code != 0 def test_missing_scene_arg_with_other_args_exits_with_error(): """--scene なしで他の引数を渡すと SystemExit が発生すること．""" with pytest.raises(SystemExit) as exc_info: main(["--no-bake"]) assert exc_info.value.code != 0 # --------------------------------------------------------------------------- # 2. 存在しない YAML パス（軽量 — return code 1） # --------------------------------------------------------------------------- def test_nonexistent_scene_returns_1(tmp_path: Path, capsys): """存在しない YAML パスを渡すと return 1 かつ stderr にエラーが出ること． 仕様: FileNotFoundError → stderr「シーン YAML ファイルが見つかりません」→ return 1 """ nonexistent = tmp_path / "does_not_exist.yaml" result = main(["--scene", str(nonexistent), "--no-bake", "--off-screen"]) assert result == 1 captured = capsys.readouterr() assert "シーン YAML ファイルが見つかりません" in captured.err def test_nonexistent_scene_error_message_is_on_stderr(tmp_path: Path, capsys): """エラーメッセージが stderr に出ること．""" nonexistent = tmp_path / "missing.yaml" main(["--scene", str(nonexistent), "--no-bake", "--off-screen"]) captured = capsys.readouterr() assert "エラー:" in captured.err # --------------------------------------------------------------------------- # 3. 不正 YAML（軽量 — return code 1） # --------------------------------------------------------------------------- def test_invalid_yaml_returns_1(invalid_yaml_file: Path, capsys): """必須フィールド欠落の YAML を渡すと return 1 が返ること． 仕様: Pydantic ValidationError → stderr「シーン YAML の読み込みに失敗しました」→ return 1 """ result = main(["--scene", str(invalid_yaml_file), "--no-bake", "--off-screen"]) assert result == 1 def test_invalid_yaml_stderr_message(invalid_yaml_file: Path, capsys): """不正 YAML のエラーメッセージに「シーン YAML の読み込みに失敗しました」が含まれること．""" main(["--scene", str(invalid_yaml_file), "--no-bake", "--off-screen"]) captured = capsys.readouterr() assert "シーン YAML の読み込みに失敗しました" in captured.err # --------------------------------------------------------------------------- # 4. --no-bake モード（slow — PyVista 実行） # --------------------------------------------------------------------------- @pytest.mark.slow def test_no_bake_with_screenshot_creates_png(tmp_path: Path): """--no-bake + --screenshot で PNG が生成されること（ベイクなし）． 仕様: --no-bake のとき bake_irradiance をスキップする """ screenshot_path = tmp_path / "no_bake_out.png" result = main([ "--scene", str(MINIMAL_YAML), "--no-bake", "--screenshot", str(screenshot_path), ]) assert result == 0, "正常終了 (return 0)" assert screenshot_path.exists(), "--no-bake + --screenshot で PNG が生成されること" assert screenshot_path.stat().st_size > 0, "PNG ファイルが空でないこと" @pytest.mark.slow def test_no_bake_does_not_print_bake_start(tmp_path: Path, capsys): """--no-bake のとき「照度ベイクを開始します」が stdout に出ないこと． 仕様: --no-bake が指定されたとき「--no-bake: 照度ベイクをスキップします」を出力する """ screenshot_path = tmp_path / "out.png" main([ "--scene", str(MINIMAL_YAML), "--no-bake", "--screenshot", str(screenshot_path), ]) captured = capsys.readouterr() assert "照度ベイクを開始します" not in captured.out assert "スキップ" in captured.out # --------------------------------------------------------------------------- # 5. --screenshot で PNG が生成されること（slow — ベイク + PyVista 実行） # --------------------------------------------------------------------------- @pytest.mark.slow def test_screenshot_creates_png(tmp_path: Path): """--screenshot で PNG が生成されること（ベイクあり）． 仕様: --screenshot PNG を保存（off_screen 強制） """ screenshot_path = tmp_path / "out.png" result = main([ "--scene", str(MINIMAL_YAML), "--screenshot", str(screenshot_path), ]) assert result == 0, "正常終了 (return 0)" assert screenshot_path.exists(), "--screenshot で PNG が生成されること" assert screenshot_path.stat().st_size > 0 @pytest.mark.slow def test_screenshot_stdout_contains_save_message(tmp_path: Path, capsys): """--screenshot 実行後に stdout に「スクリーンショット保存」が含まれること． 仕様: screenshot 指定時は完了後に「スクリーンショット保存: 」を出力する """ screenshot_path = tmp_path / "out.png" main([ "--scene", str(MINIMAL_YAML), "--screenshot", str(screenshot_path), ]) captured = capsys.readouterr() assert "スクリーンショット保存" in captured.out # --------------------------------------------------------------------------- # 6. --output でベイク結果 npz が保存されること（slow） # --------------------------------------------------------------------------- @pytest.mark.slow def test_output_dir_creates_npz(tmp_path: Path): """--output 指定でベイク結果 npz が保存されること． 仕様: --output / -o ベイク結果を {config.name}_irradiance.npz として保存 """ output_dir = tmp_path / "bake_out" screenshot_path = tmp_path / "out.png" result = main([ "--scene", str(MINIMAL_YAML), "--output", str(output_dir), "--screenshot", str(screenshot_path), ]) assert result == 0 # {config.name}_irradiance.npz が存在すること # minimal.yaml の name: smoke_test npz_path = output_dir / "smoke_test_irradiance.npz" assert npz_path.exists(), f"npz ファイルが生成されること: {npz_path}" # npz が正しくロードできること data = np.load(str(npz_path)) assert len(data.files) > 0, "npz に少なくとも 1 つのキーが含まれること" @pytest.mark.slow def test_output_dir_is_created_automatically(tmp_path: Path): """--output で指定したディレクトリが自動生成されること． 仕様: output_dir.mkdir(parents=True, exist_ok=True) で親ディレクトリも含めて作成する """ output_dir = tmp_path / "deep" / "nested" / "bake" screenshot_path = tmp_path / "out.png" assert not output_dir.exists(), "テスト前にはディレクトリが存在しないこと" result = main([ "--scene", str(MINIMAL_YAML), "--output", str(output_dir), "--screenshot", str(screenshot_path), ]) assert result == 0 assert output_dir.exists(), "深いネストのディレクトリが自動作成されること" # --------------------------------------------------------------------------- # 7. 短縮引数 -s / -o（slow） # --------------------------------------------------------------------------- @pytest.mark.slow def test_short_args_s_and_o_work(tmp_path: Path): """-s と -o の短縮形でも --scene / --output と同等に動くこと． 仕様: --scene/-s, --output/-o の短縮形を提供する """ output_dir = tmp_path / "short_out" screenshot_path = tmp_path / "out.png" result = main([ "-s", str(MINIMAL_YAML), "-o", str(output_dir), "--screenshot", str(screenshot_path), ]) assert result == 0 assert (output_dir / "smoke_test_irradiance.npz").exists() # --------------------------------------------------------------------------- # 8. --off-screen + ベイクのみ（screenshot なし，no-op）（slow） # --------------------------------------------------------------------------- @pytest.mark.slow def test_off_screen_without_screenshot_returns_0(tmp_path: Path): """--off-screen + screenshot なしで no-op になり return 0 が返ること． 仕様: screenshot=None かつ off_screen=True のときは no-op """ result = main([ "--scene", str(MINIMAL_YAML), "--no-bake", "--off-screen", ]) assert result == 0, "no-op でも正常終了すること"