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>
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
|
|
np.testing.assert_allclose(new_vmobject.points[:-4], test_input.points)
|