Class Test_PythonParser

tests/test_parser/test_python.py:10–147

class Test_PythonParser(unittest.TestCase):
    def setUp(self) -> None:        
        with open('tests/test_parser/test_sample/py_test_sample.py', 'r') as file:
            self.code_sample = file.read()
        
        tree = parse_code(self.code_sample, 'python')
        self.root_node = tree.root_node
        return super().setUp()

    def test_get_function_list(self):
        root = self.root_node
        
        function_list = PythonParser.get_function_list(root)
        
        self.assertEqual(len(function_list), 3)

    def test_get_class_list(self):
        root = self.root_node
        
        class_list = PythonParser.get_class_list(root)
        self.assertEqual(len(class_list), 1)

    def test_get_docstring(self):
        code_sample = '''
        def test_sample():
            """This is a docstring"""
            return
        '''
        root = parse_code(code_sample, 'python').root_node
        
        function = PythonParser.get_function_list(root)[0]
        docstring = PythonParser.get_docstring(function)
        self.assertEqual(docstring, "This is a docstring")

    def test_get_function_metadata(self):
        code_sample = '''
        def test_sample(arg1: str = "string", arg2 = "another_string"):
            return NotImplement()
        '''
        root = parse_code(code_sample, 'python').root_node
        
        function = list(PythonParser.get_function_list(root))[0]
        metadata = PythonParser.get_function_metadata(function)

        for key in ['identifier', 'parameters', 'return_type']:
            self.assertTrue(key in metadata.keys())
        self.assertEqual(metadata['parameters'], {'arg1': 'str', 'arg2': None})
        self.assertEqual(metadata['identifier'], 'test_sample')

    def test_get_class_metadata(self):
        code_sample = '''
        class ABC():
            pass
            
        class Sample(ABC):
            def __init__(self):
                pass