Method testFunction

python/tests/test_convert_ast.py:82–174

(self)

        self.assertEqual(fileNode, HELLO_NAME_SPLOOT)

    def testFunction(self):
        self.maxDiff = None
        fileNode = splootFromPython('''
def add(a, b):
    print(a + b)
add(3, 4)
''')
        self.assertEqual(fileNode, {
"type":"PYTHON_FILE",
"properties":{},
"childSets":{
    "body": [
        {
            "type": "PYTHON_STATEMENT",
            "meta": {"lineno": 1},
            "childSets": {
                "statement": [{
                    "type": "PYTHON_FUNCTION_DECLARATION",
                    "properties": {'id': None},
                    "childSets": {
                        "decorators": [],
                        "identifier": [{'type': 'PY_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'add'}}],
                        "body": [
                            {
                                "type": "PYTHON_STATEMENT",
                                "meta": {"lineno": 2},
                                "childSets": {
                                    "statement": [{
                                        "type": "PYTHON_EXPRESSION",
                                        "properties": {},
                                        "childSets": {
                                            "tokens": [
                                                {"type": "PYTHON_CALL_VARIABLE",
                                                'properties': {'identifier': 'print'},
                                                "childSets": {
                                                    "arguments": [
                                                        {"type": "PY_ARG", "properties": {}, "childSets": {"argument": [
                                                            {"type": "PYTHON_EXPRESSION", "properties": {}, "childSets": {'tokens': [
                                                                {'type': 'PY_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'a'}},
                                                                {'type': 'PYTHON_BINARY_OPERATOR', 'childSets': {}, 'properties': {'operator': '+'}},
                                                                {'type': 'PY_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'b'}},
                                                            ]}},
                                                        ]}},
                                                    ]
                                                }}
                                            ],
                                        }
                                    }]
                                },
                                "properties": {}
                            }
                        ],
                        "params": [
                            {'type': 'PY_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'a'}},
                            {'type': 'PY_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'b'}}
                        ]
                    }
                }]