Method testImportFrom

python/tests/test_convert_ast.py:263–290

(self)

    def testImportFrom(self):
        self.maxDiff = None

        fileNode = splootFromPython('from random import randint')
        self.assertEqual(fileNode, {
            'type': 'PYTHON_FILE',
            'childSets': {'body': [
                {
                    "type": "PYTHON_STATEMENT",
                    "childSets": {
                        "statement": [{
                            'type': 'PYTHON_FROM_IMPORT',
                            'childSets': {
                                'attrs': [
                                    {'type': 'PY_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'randint'}}
                                ],
                                'module': [
                                    {'type': 'PYTHON_MODULE_IDENTIFIER', 'childSets': {}, 'properties': {'identifier': 'random'}}
                                ]
                            },
                            'properties': {},
                        }]
                    },
                    'properties': {},
                }
            ]},
            'properties': {},
        })

    def testTuple(self):
        self.maxDiff = None