Method test_basic_cases

codegen/test_missing_colons.py:21–33

(self)

                    self.assertTrue(self.check_syntax(fixed_code), msg=f"\n\nFailed test {i} in {test_type}. Syntax checker found a problem with the output: {fixed_code}")

    def test_basic_cases(self):
        test_cases = [
            ("def foo(x)\n    pass", "def foo(x):\n    pass"),
            ("if x > 0\n    pass\nelif x < 0\n    pass", "if x > 0:\n    pass\nelif x < 0:\n    pass"),
            ("if x > 0\n    pass\nelse\n    pass", "if x > 0:\n    pass\nelse:\n    pass"),
            ("for i in range(10)\n    pass", "for i in range(10):\n    pass"),
            ("while True\n    pass", "while True:\n    pass"),
            ("with open('file.txt') as f\n    pass", "with open('file.txt') as f:\n    pass"),
            ("class MyClass\n    def f():\n        pass", "class MyClass:\n    def f():\n        pass"),
            ("print(f\"The area of a circle with radius {radius} is {area:.2f}\")", "print(f\"The area of a circle with radius {radius} is {area:.2f}\")")
        ]

        self.run_test_cases(test_cases, "Basic", syntax_check=False)

    def test_single_line(self):
        test_cases = [