| 6 | |
| 7 | |
| 8 | def test_code_initialization_from_string(): |
| 9 | code_string = """from manim import Scene, Square |
| 10 | |
| 11 | class FadeInSquare(Scene): |
| 12 | def construct(self): |
| 13 | s = Square() |
| 14 | self.play(FadeIn(s)) |
| 15 | self.play(s.animate.scale(2)) |
| 16 | self.wait()""" |
| 17 | rendered_code = Code( |
| 18 | code_string=code_string, |
| 19 | language="python", |
| 20 | ) |
| 21 | num_lines = len(code_string.split("\n")) |
| 22 | assert len(rendered_code.code_lines) == num_lines |
| 23 | assert len(rendered_code.line_numbers) == num_lines |
| 24 | |
| 25 | |
| 26 | def test_code_initialization_from_file(): |