gazelle/python/parse_test.py

import unittest

import parse


class TestParse(unittest.TestCase):
    def test_not_has_main(self):
        content = "a = 1\nb = 2"
        self.assertFalse(parse.parse_main(content))

    def test_has_main_in_function(self):
        content = """
def foo():
    if __name__ == "__main__":
        a = 3
"""
        self.assertFalse(parse.parse_main(content))

    def test_has_main(self):
        content = """
import unittest

from lib import main


class ExampleTest(unittest.TestCase):
    def test_main(self):
        self.assertEqual(
            "",
            main([["A", 1], ["B", 2]]),
        )


if __name__ == "__main__":
    unittest.main()
"""
        self.assertTrue(parse.parse_main(content))


if __name__ == "__main__":
    unittest.main()