mirror of
https://github.com/ManimCommunity/manim.git
synced 2026-06-22 10:01:47 +00:00
106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from manim import capture
|
|
from tests.assert_utils import assert_dir_exists, assert_dir_not_exists
|
|
|
|
from ..utils.video_tester import video_comparison
|
|
|
|
|
|
@pytest.mark.slow
|
|
@video_comparison(
|
|
"SceneWithDisabledSections.json",
|
|
"videos/simple_scenes/480p15/SquareToCircle.mp4",
|
|
)
|
|
def test_no_sections(tmp_path, manim_cfg_file, simple_scenes_path):
|
|
scene_name = "SquareToCircle"
|
|
command = [
|
|
sys.executable,
|
|
"-m",
|
|
"manim",
|
|
"-ql",
|
|
"--media_dir",
|
|
str(tmp_path),
|
|
simple_scenes_path,
|
|
scene_name,
|
|
]
|
|
_, err, exit_code = capture(command)
|
|
assert exit_code == 0, err
|
|
|
|
scene_dir = os.path.join(tmp_path, "videos", "simple_scenes", "480p15")
|
|
assert_dir_exists(scene_dir)
|
|
assert_dir_not_exists(os.path.join(scene_dir, "sections"))
|
|
|
|
|
|
@pytest.mark.slow
|
|
@video_comparison(
|
|
"SceneWithEnabledSections.json",
|
|
"videos/simple_scenes/480p15/SquareToCircle.mp4",
|
|
)
|
|
def test_sections(tmp_path, manim_cfg_file, simple_scenes_path):
|
|
scene_name = "SquareToCircle"
|
|
command = [
|
|
sys.executable,
|
|
"-m",
|
|
"manim",
|
|
"-ql",
|
|
"--save_sections",
|
|
"--media_dir",
|
|
str(tmp_path),
|
|
simple_scenes_path,
|
|
scene_name,
|
|
]
|
|
_, err, exit_code = capture(command)
|
|
assert exit_code == 0, err
|
|
|
|
scene_dir = os.path.join(tmp_path, "videos", "simple_scenes", "480p15")
|
|
assert_dir_exists(scene_dir)
|
|
assert_dir_exists(os.path.join(scene_dir, "sections"))
|
|
|
|
|
|
@pytest.mark.slow
|
|
@video_comparison(
|
|
"SceneWithSections.json",
|
|
"videos/simple_scenes/480p15/SceneWithSections.mp4",
|
|
)
|
|
def test_many_sections(tmp_path, manim_cfg_file, simple_scenes_path):
|
|
scene_name = "SceneWithSections"
|
|
command = [
|
|
sys.executable,
|
|
"-m",
|
|
"manim",
|
|
"-ql",
|
|
"--save_sections",
|
|
"--media_dir",
|
|
str(tmp_path),
|
|
simple_scenes_path,
|
|
scene_name,
|
|
]
|
|
_, err, exit_code = capture(command)
|
|
assert exit_code == 0, err
|
|
|
|
|
|
@pytest.mark.slow
|
|
@video_comparison(
|
|
"SceneWithSkipAnimations.json",
|
|
"videos/simple_scenes/480p15/ElaborateSceneWithSections.mp4",
|
|
)
|
|
def test_skip_animations(tmp_path, manim_cfg_file, simple_scenes_path):
|
|
scene_name = "ElaborateSceneWithSections"
|
|
command = [
|
|
sys.executable,
|
|
"-m",
|
|
"manim",
|
|
"-ql",
|
|
"--save_sections",
|
|
"--media_dir",
|
|
str(tmp_path),
|
|
simple_scenes_path,
|
|
scene_name,
|
|
]
|
|
_, err, exit_code = capture(command)
|
|
assert exit_code == 0, err
|