mirror of
https://github.com/ManimCommunity/manim.git
synced 2026-06-22 10:01:47 +00:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from manim import Circle, Square
|
|
from manim.mobject.geometry.boolean_ops import _BooleanOps
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_input,expected",
|
|
[
|
|
(
|
|
[(1.0, 2.0), (3.0, 4.0)],
|
|
[
|
|
np.array([1.0, 2.0, 0]),
|
|
np.array([3.0, 4.0, 0]),
|
|
],
|
|
),
|
|
(
|
|
[(1.1, 2.2)],
|
|
[
|
|
np.array([1.1, 2.2, 0.0]),
|
|
],
|
|
),
|
|
],
|
|
)
|
|
def test_convert_2d_to_3d_array(test_input, expected):
|
|
a = _BooleanOps()
|
|
result = a._convert_2d_to_3d_array(test_input)
|
|
assert len(result) == len(expected)
|
|
for i in range(len(result)):
|
|
assert (result[i] == expected[i]).all()
|
|
|
|
|
|
def test_convert_2d_to_3d_array_zdim():
|
|
a = _BooleanOps()
|
|
result = a._convert_2d_to_3d_array([(1.0, 2.0)], z_dim=1.0)
|
|
assert (result[0] == np.array([1.0, 2.0, 1.0])).all()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_input",
|
|
[
|
|
Square(),
|
|
Circle(),
|
|
Square(side_length=4),
|
|
Circle(radius=3),
|
|
],
|
|
)
|
|
def test_vmobject_to_skia_path_and_inverse(test_input):
|
|
a = _BooleanOps()
|
|
path = a._convert_vmobject_to_skia_path(test_input)
|
|
assert len(list(path.segments)) > 1
|
|
|
|
new_vmobject = a._convert_skia_path_to_vmobject(path)
|
|
# for some reason there is an extra 4 points in new vmobject than original
|
|
assert np.where(np.isclose(new_vmobject.points[:-4], test_input.points))
|