mirror of
https://github.com/ManimCommunity/manim.git
synced 2026-06-22 10:01:47 +00:00
677e88ef96* start test reorganization * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * move boolean ops * move more tests * Move more tests * more reorganization * more movements * mostly finish up reorganization * Finish reorganization tests in parent dir * try moving interface tests back * Move test_commands over * fix relative path to templates Co-authored-by: Darylgolden <darylgolden@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
122 lines
3.7 KiB
Python
122 lines
3.7 KiB
Python
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from manim.animation.creation import Uncreate
|
|
from manim.mobject.geometry.arc import Dot
|
|
from manim.mobject.geometry.line import Line
|
|
from manim.mobject.geometry.polygram import Square
|
|
from manim.mobject.mobject import override_animate
|
|
from manim.mobject.types.vectorized_mobject import VGroup
|
|
|
|
|
|
def test_simple_animate():
|
|
s = Square()
|
|
scale_factor = 2
|
|
anim = s.animate.scale(scale_factor).build()
|
|
assert anim.mobject.target.width == scale_factor * s.width
|
|
|
|
|
|
def test_chained_animate():
|
|
s = Square()
|
|
scale_factor = 2
|
|
direction = np.array((1, 1, 0))
|
|
anim = s.animate.scale(scale_factor).shift(direction).build()
|
|
assert (
|
|
anim.mobject.target.width == scale_factor * s.width
|
|
and (anim.mobject.target.get_center() == direction).all()
|
|
)
|
|
|
|
|
|
def test_overridden_animate():
|
|
class DotsWithLine(VGroup):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.left_dot = Dot().shift((-1, 0, 0))
|
|
self.right_dot = Dot().shift((1, 0, 0))
|
|
self.line = Line(self.left_dot, self.right_dot)
|
|
self.add(self.left_dot, self.right_dot, self.line)
|
|
|
|
def remove_line(self):
|
|
self.remove(self.line)
|
|
|
|
@override_animate(remove_line)
|
|
def _remove_line_animation(self, anim_args=None):
|
|
if anim_args is None:
|
|
anim_args = {}
|
|
self.remove_line()
|
|
return Uncreate(self.line, **anim_args)
|
|
|
|
dots_with_line = DotsWithLine()
|
|
anim = dots_with_line.animate.remove_line().build()
|
|
assert len(dots_with_line.submobjects) == 2
|
|
assert type(anim) is Uncreate
|
|
|
|
|
|
def test_chaining_overridden_animate():
|
|
class DotsWithLine(VGroup):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.left_dot = Dot().shift((-1, 0, 0))
|
|
self.right_dot = Dot().shift((1, 0, 0))
|
|
self.line = Line(self.left_dot, self.right_dot)
|
|
self.add(self.left_dot, self.right_dot, self.line)
|
|
|
|
def remove_line(self):
|
|
self.remove(self.line)
|
|
|
|
@override_animate(remove_line)
|
|
def _remove_line_animation(self, anim_args=None):
|
|
if anim_args is None:
|
|
anim_args = {}
|
|
self.remove_line()
|
|
return Uncreate(self.line, **anim_args)
|
|
|
|
with pytest.raises(
|
|
NotImplementedError,
|
|
match="not supported for overridden animations",
|
|
):
|
|
DotsWithLine().animate.shift((1, 0, 0)).remove_line()
|
|
|
|
with pytest.raises(
|
|
NotImplementedError,
|
|
match="not supported for overridden animations",
|
|
):
|
|
DotsWithLine().animate.remove_line().shift((1, 0, 0))
|
|
|
|
|
|
def test_animate_with_args():
|
|
s = Square()
|
|
scale_factor = 2
|
|
run_time = 2
|
|
|
|
anim = s.animate(run_time=run_time).scale(scale_factor).build()
|
|
assert anim.mobject.target.width == scale_factor * s.width
|
|
assert anim.run_time == run_time
|
|
|
|
|
|
def test_chained_animate_with_args():
|
|
s = Square()
|
|
scale_factor = 2
|
|
direction = np.array((1, 1, 0))
|
|
run_time = 2
|
|
|
|
anim = s.animate(run_time=run_time).scale(scale_factor).shift(direction).build()
|
|
assert (
|
|
anim.mobject.target.width == scale_factor * s.width
|
|
and (anim.mobject.target.get_center() == direction).all()
|
|
)
|
|
assert anim.run_time == run_time
|
|
|
|
|
|
def test_animate_with_args_misplaced():
|
|
s = Square()
|
|
scale_factor = 2
|
|
run_time = 2
|
|
|
|
with pytest.raises(ValueError, match="must be passed before"):
|
|
s.animate.scale(scale_factor)(run_time=run_time)
|
|
|
|
with pytest.raises(ValueError, match="must be passed before"):
|
|
s.animate(run_time=run_time)(run_time=run_time).scale(scale_factor)
|