from __future__ import annotations import sysfrom pathlib import Path import numpy as npimport pytestfrom click.testing import CliRunnerfrom PIL import Image from manim import capture, get_video_metadatafrom manim.__main__ import __version__, mainfrom manim.utils.file_ops import add_version_before_extension from ..utils.video_tester import video_comparison  @pytest.mark.slow@video_comparison(    "SquareToCircleWithDefaultValues.json",    "videos/simple_scenes/1080p60/SquareToCircle.mp4",)def test_basic_scene_with_default_values(tmp_path, manim_cfg_file, simple_scenes_path):    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "--media_dir",        str(tmp_path),        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err  @pytest.mark.slowdef test_resolution_flag(tmp_path, manim_cfg_file, simple_scenes_path):    scene_name = "NoAnimations"    # test different separators    resolutions = [        (720, 480, ";"),        (1280, 720, ","),        (1920, 1080, "-"),        (2560, 1440, ";"),        # (3840, 2160, ","),        # (640, 480, "-"),        # (800, 600, ";"),    ]     for width, height, separator in resolutions:        command = [            sys.executable,            "-m",            "manim",            "--media_dir",            str(tmp_path),            "--resolution",            f"{width}{separator}{height}",            str(simple_scenes_path),            scene_name,        ]         _, err, exit_code = capture(command)        assert exit_code == 0, err         path = (            tmp_path / "videos" / "simple_scenes" / f"{height}p60" / f"{scene_name}.mp4"        )        meta = get_video_metadata(path)        assert (width, height) == (meta["width"], meta["height"])  @pytest.mark.slow@video_comparison(    "SquareToCircleWithlFlag.json",    "videos/simple_scenes/480p15/SquareToCircle.mp4",)def test_basic_scene_l_flag(tmp_path, manim_cfg_file, simple_scenes_path):    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err  @pytest.mark.slow@video_comparison(    "SceneWithMultipleCallsWithNFlag.json",    "videos/simple_scenes/480p15/SceneWithMultipleCalls.mp4",)def test_n_flag(tmp_path, simple_scenes_path):    scene_name = "SceneWithMultipleCalls"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "-n 3,6",        "--media_dir",        str(tmp_path),        str(simple_scenes_path),        scene_name,    ]    _, err, exit_code = capture(command)    assert exit_code == 0, err  @pytest.mark.slowdef test_s_flag_no_animations(tmp_path, manim_cfg_file, simple_scenes_path):    scene_name = "NoAnimations"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "-s",        "--media_dir",        str(tmp_path),        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     exists = (tmp_path / "videos").exists()    assert not exists, "running manim with -s flag rendered a video"     is_empty = not any((tmp_path / "images" / "simple_scenes").iterdir())    assert not is_empty, "running manim with -s flag did not render an image"  @pytest.mark.slow@pytest.mark.parametrize("still_flag", ["-s", "--format=png"])def test_s_flag(tmp_path, manim_cfg_file, simple_scenes_path, still_flag):    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        still_flag,        "--media_dir",        str(tmp_path),        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     exists = (tmp_path / "videos").exists()    assert not exists, "running manim with -s flag rendered a video"     is_empty = not any((tmp_path / "images" / "simple_scenes").iterdir())    assert not is_empty, "running manim with -s flag did not render an image"  @pytest.mark.slowdef test_s_flag_opengl_renderer(tmp_path, manim_cfg_file, simple_scenes_path):    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "-s",        "--renderer",        "opengl",        "--media_dir",        str(tmp_path),        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     exists = (tmp_path / "videos").exists()    assert not exists, "running manim with -s flag rendered a video"     is_empty = not any((tmp_path / "images" / "simple_scenes").iterdir())    assert not is_empty, "running manim with -s flag did not render an image"  @pytest.mark.slowdef test_r_flag(tmp_path, manim_cfg_file, simple_scenes_path):    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "-s",        "--media_dir",        str(tmp_path),        "-r",        "200,100",        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     is_not_empty = any((tmp_path / "images").iterdir())    assert is_not_empty, "running manim with -s, -r flag did not render a file"     filename = add_version_before_extension(        tmp_path / "images" / "simple_scenes" / "SquareToCircle.png",    )    assert np.asarray(Image.open(filename)).shape == (100, 200, 4)  @pytest.mark.slowdef test_a_flag(tmp_path, manim_cfg_file, infallible_scenes_path):    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        "-a",        str(infallible_scenes_path),    ]    _, err, exit_code = capture(command)    assert exit_code == 0, err     one_is_not_empty = (        tmp_path / "videos" / "infallible_scenes" / "480p15" / "Wait1.mp4"    ).is_file()    assert one_is_not_empty, "running manim with -a flag did not render the first scene"     two_is_not_empty = (        tmp_path / "images" / "infallible_scenes" / f"Wait2_ManimCE_v{__version__}.png"    ).is_file()    assert two_is_not_empty, (        "running manim with -a flag did not render an image, possible leak of the config dictionary."    )     three_is_not_empty = (        tmp_path / "videos" / "infallible_scenes" / "480p15" / "Wait3.mp4"    ).is_file()    assert three_is_not_empty, (        "running manim with -a flag did not render the second scene"    )  @pytest.mark.parametrize(    "scene_selection",    [        ("-a",),        ("Wait1", "Wait3"),    ],)def test_output_file_rejects_multi_scene_render(    tmp_path,    infallible_scenes_path,    scene_selection,):    command = [        sys.executable,        "-m",        "manim",        "--media_dir",        str(tmp_path),        "-o",        "shared-name",    ]    if scene_selection == ("-a",):        command.extend(["-a", str(infallible_scenes_path)])    else:        command.extend([str(infallible_scenes_path), *scene_selection])     out, err, exit_code = capture(command)     assert exit_code == 1    assert "--output_file can only be used when rendering exactly one scene" in (        err or out    )    assert not tmp_path.exists() or not any(tmp_path.iterdir())  @pytest.mark.slowdef test_custom_output_name_gif(tmp_path, simple_scenes_path):    scene_name = "SquareToCircle"    custom_name = "custom_name"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        "--format=gif",        "-o",        custom_name,        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     wrong_gif_path = add_version_before_extension(        tmp_path / "videos" / "simple_scenes" / "480p15" / f"{scene_name}.gif"    )     assert not wrong_gif_path.exists(), (        "The gif file does not respect the custom name: " + custom_name + ".gif"    )     unexpected_mp4_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / str(custom_name + ".mp4")    )     assert not unexpected_mp4_path.exists(), "Found an unexpected mp4 file at " + str(        unexpected_mp4_path    )     expected_gif_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / str(custom_name + ".gif")    )     assert expected_gif_path.exists(), "gif file not found at " + str(expected_gif_path)  @pytest.mark.slowdef test_custom_output_name_mp4(tmp_path, simple_scenes_path):    scene_name = "SquareToCircle"    requested_name = "custom_name.mov"    expected_name = requested_name    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        "-o",        requested_name,        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     wrong_mp4_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / str(scene_name + ".mp4")    )     assert not wrong_mp4_path.exists(), (        "The mp4 file does not respect the custom name: " + expected_name + ".mp4"    )     unexpected_gif_path = add_version_before_extension(        tmp_path / "videos" / "simple_scenes" / "480p15" / f"{expected_name}.gif"    )    assert not unexpected_gif_path.exists(), "Found an unexpected gif file at " + str(        unexpected_gif_path    )     expected_mp4_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / str(expected_name + ".mp4")    )     assert expected_mp4_path.exists(), "mp4 file not found at " + str(expected_mp4_path)  @pytest.mark.slowdef test_dash_as_filename(tmp_path):    code = (        "class Test(Scene):\n"        "    def construct(self):\n"        "        self.add(Circle())\n"        "        self.wait()"    )    command = [        "-ql",        "-s",        "--media_dir",        str(tmp_path),        "-",    ]    runner = CliRunner()    result = runner.invoke(main, command, input=code)    assert result.exit_code == 0    exists = add_version_before_extension(        tmp_path / "images" / "-" / "Test.png",    ).exists()    assert exists, result.output  @pytest.mark.slowdef test_gif_format_output(tmp_path, manim_cfg_file, simple_scenes_path):    """Test only gif created with manim version in file name when --format gif is set"""    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        "--format",        "gif",        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     unexpected_mp4_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / "SquareToCircle.mp4"    )    assert not unexpected_mp4_path.exists(), "unexpected mp4 file found at " + str(        unexpected_mp4_path,    )     expected_gif_path = add_version_before_extension(        tmp_path / "videos" / "simple_scenes" / "480p15" / "SquareToCircle.gif"    )    assert expected_gif_path.exists(), "gif file not found at " + str(expected_gif_path)  @pytest.mark.slowdef test_mp4_format_output(tmp_path, manim_cfg_file, simple_scenes_path):    """Test only mp4 created without manim version in file name when --format mp4 is set"""    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        "--format",        "mp4",        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     unexpected_gif_path = add_version_before_extension(        tmp_path / "videos" / "simple_scenes" / "480p15" / "SquareToCircle.gif"    )    assert not unexpected_gif_path.exists(), "unexpected gif file found at " + str(        unexpected_gif_path,    )     expected_mp4_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / "SquareToCircle.mp4"    )    assert expected_mp4_path.exists(), "expected mp4 file not found at " + str(        expected_mp4_path,    )  @pytest.mark.slowdef test_videos_not_created_when_png_format_set(    tmp_path,    manim_cfg_file,    simple_scenes_path,):    """Test mp4 and gifs are not created when --format png is set"""    scene_name = "SquareToCircle"    command = [        sys.executable,        "-m",        "manim",        "-ql",        "--media_dir",        str(tmp_path),        "--format",        "png",        str(simple_scenes_path),        scene_name,    ]    out, err, exit_code = capture(command)    assert exit_code == 0, err     unexpected_gif_path = add_version_before_extension(        tmp_path / "videos" / "simple_scenes" / "480p15" / "SquareToCircle.gif"    )    assert not unexpected_gif_path.exists(), "unexpected gif file found at " + str(        unexpected_gif_path,    )     unexpected_mp4_path = (        tmp_path / "videos" / "simple_scenes" / "480p15" / "SquareToCircle.mp4"    )    assert not unexpected_mp4_path.exists(), "expected mp4 file not found at " + str(        unexpected_mp4_path,

Lines 1–500 of 839. Load more