mirror of
https://github.com/ManimCommunity/manim.git
synced 2026-06-22 10:01:47 +00:00
3e8f41c9be* fixed problem: default value of color in styles in CodeMobject * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * added accidently removed test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update code_mobject.py Code suggested by maejam --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Henrik Skov Midtiby <hemi@mmmi.sdu.dk> Co-authored-by: Henrik Skov Midtiby <henrikmidtiby@gmail.com>
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
from manim.mobject.text.code_mobject import Code
|
|
from manim.utils.color.core import ManimColor
|
|
|
|
|
|
def test_code_initialization_from_string():
|
|
code_string = """from manim import Scene, Square
|
|
|
|
class FadeInSquare(Scene):
|
|
def construct(self):
|
|
s = Square()
|
|
self.play(FadeIn(s))
|
|
self.play(s.animate.scale(2))
|
|
self.wait()"""
|
|
rendered_code = Code(
|
|
code_string=code_string,
|
|
language="python",
|
|
)
|
|
num_lines = len(code_string.split("\n"))
|
|
assert len(rendered_code.code_lines) == num_lines
|
|
assert len(rendered_code.line_numbers) == num_lines
|
|
|
|
|
|
def test_code_initialization_from_file():
|
|
rendered_code = Code(
|
|
code_file="tests/test_code_mobject.py",
|
|
language="python",
|
|
background="window",
|
|
background_config={"fill_color": "#101010"},
|
|
)
|
|
assert len(rendered_code.code_lines) == len(rendered_code.line_numbers)
|
|
assert rendered_code.background.fill_color == ManimColor("#101010")
|
|
|
|
|
|
def test_line_heights_initial_whitespace():
|
|
rendered_code = Code(
|
|
code_string="""print('Hello, World!')
|
|
for _ in range(42):
|
|
print('Hello, World!')
|
|
""",
|
|
language="python",
|
|
)
|
|
np.testing.assert_almost_equal(
|
|
rendered_code.code_lines[0].height,
|
|
rendered_code.code_lines[2].height,
|
|
)
|
|
|
|
|
|
def test_code_initialization_style_correct_color():
|
|
for style in Code.get_styles_list():
|
|
try:
|
|
Code(
|
|
code_string="""# This is a comment.
|
|
var = 3
|
|
print(var)
|
|
""",
|
|
formatter_style=style,
|
|
)
|
|
except ValueError as e:
|
|
pytest.fail(f"Code initialization failed for style {style} with error: {e}")
|