mirror of
https://github.com/ManimCommunity/manim.git
synced 2026-06-22 10:01:47 +00:00
120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
import pytest
|
|
from manim import Mobject, VMobject, VGroup, VDict
|
|
|
|
|
|
def test_vgroup_init():
|
|
"""Test the VGroup instantiation."""
|
|
VGroup()
|
|
VGroup(VMobject())
|
|
VGroup(VMobject(), VMobject())
|
|
with pytest.raises(TypeError):
|
|
VGroup(Mobject())
|
|
with pytest.raises(TypeError):
|
|
VGroup(Mobject(), Mobject())
|
|
|
|
|
|
def test_vgroup_add():
|
|
"""Test the VGroup add method."""
|
|
obj = VGroup()
|
|
assert len(obj.submobjects) == 0
|
|
obj.add(VMobject())
|
|
assert len(obj.submobjects) == 1
|
|
with pytest.raises(TypeError):
|
|
obj.add(Mobject())
|
|
assert len(obj.submobjects) == 1
|
|
with pytest.raises(TypeError):
|
|
# If only one of the added object is not an instance of VMobject, none of them should be added
|
|
obj.add(VMobject(), Mobject())
|
|
assert len(obj.submobjects) == 1
|
|
with pytest.raises(Exception): # TODO change this to ValueError once #307 is merged
|
|
# a Mobject cannot contain itself
|
|
obj.add(obj)
|
|
|
|
|
|
def test_vgroup_add_dunder():
|
|
"""Test the VGroup __add__ magic method."""
|
|
obj = VGroup()
|
|
assert len(obj.submobjects) == 0
|
|
obj + VMobject()
|
|
assert len(obj.submobjects) == 0
|
|
obj += VMobject()
|
|
assert len(obj.submobjects) == 1
|
|
with pytest.raises(TypeError):
|
|
obj += Mobject()
|
|
assert len(obj.submobjects) == 1
|
|
with pytest.raises(TypeError):
|
|
# If only one of the added object is not an instance of VMobject, none of them should be added
|
|
obj += (VMobject(), Mobject())
|
|
assert len(obj.submobjects) == 1
|
|
with pytest.raises(Exception): # TODO change this to ValueError once #307 is merged
|
|
# a Mobject cannot contain itself
|
|
obj += obj
|
|
|
|
|
|
def test_vgroup_remove():
|
|
"""Test the VGroup remove method."""
|
|
a = VMobject()
|
|
c = VMobject()
|
|
b = VGroup(c)
|
|
obj = VGroup(a, b)
|
|
assert len(obj.submobjects) == 2
|
|
assert len(b.submobjects) == 1
|
|
obj.remove(a)
|
|
b.remove(c)
|
|
assert len(obj.submobjects) == 1
|
|
assert len(b.submobjects) == 0
|
|
obj.remove(b)
|
|
assert len(obj.submobjects) == 0
|
|
|
|
|
|
def test_vgroup_remove_dunder():
|
|
"""Test the VGroup __sub__ magic method."""
|
|
a = VMobject()
|
|
c = VMobject()
|
|
b = VGroup(c)
|
|
obj = VGroup(a, b)
|
|
assert len(obj.submobjects) == 2
|
|
assert len(b.submobjects) == 1
|
|
assert len((obj - a)) == 1
|
|
assert len(obj.submobjects) == 2
|
|
obj -= a
|
|
b -= c
|
|
assert len(obj.submobjects) == 1
|
|
assert len(b.submobjects) == 0
|
|
obj -= b
|
|
assert len(obj.submobjects) == 0
|
|
|
|
|
|
def test_vdict_init():
|
|
"""Test the VDict instantiation."""
|
|
# Test empty VDict
|
|
VDict()
|
|
# Test VDict made from list of pairs
|
|
VDict([("a", VMobject()), ("b", VMobject()), ("c", VMobject())])
|
|
# Test VDict made from a python dict
|
|
VDict({"a": VMobject(), "b": VMobject(), "c": VMobject()})
|
|
# Test VDict made using zip
|
|
VDict(zip(["a", "b", "c"], [VMobject(), VMobject(), VMobject()]))
|
|
# If the value is of type Mobject, must raise a TypeError
|
|
with pytest.raises(TypeError):
|
|
VDict({"a": Mobject()})
|
|
|
|
|
|
def test_vdict_add():
|
|
"""Test the VDict add method."""
|
|
obj = VDict()
|
|
assert len(obj.submob_dict) == 0
|
|
obj.add([("a", VMobject())])
|
|
assert len(obj.submob_dict) == 1
|
|
with pytest.raises(TypeError):
|
|
obj.add([("b", Mobject())])
|
|
|
|
|
|
def test_vdict_remove():
|
|
"""Test the VDict remove method."""
|
|
obj = VDict([("a", VMobject())])
|
|
assert len(obj.submob_dict) == 1
|
|
obj.remove("a")
|
|
assert len(obj.submob_dict) == 0
|
|
with pytest.raises(KeyError):
|
|
obj.remove("a")
|