Python get_nodes_by_instance_type Examples

Example #1

File: source_parser_unittest.py Project: 3ancho/Gooey

  def test_parser_identifies_import_module(self):
    source = '''
import os
import itertools
from os import path
    '''
    import _ast
    nodes = ast.parse(source)
    module_imports = source_parser.get_nodes_by_instance_type(nodes, _ast.Import)
    self.assertEqual(2, len(module_imports))

Example #2

File: source_parser_unittest.py Project: 3ancho/Gooey

  def test_parser_identifies_import_from(self):
    source = '''
import os
import itertools
from os import path
from gooey.gooey_decorator import Gooey
    '''
    import _ast
    nodes = ast.parse(source)
    from_imports = source_parser.get_nodes_by_instance_type(nodes, _ast.ImportFrom)
    self.assertEqual(2, len(from_imports))

Example #3

    def test_parser_identifies_import_module(self):
        source = '''
import os
import itertools
from os import path
    '''
        import _ast
        nodes = ast.parse(source)
        module_imports = source_parser.get_nodes_by_instance_type(
            nodes, _ast.Import)
        self.assertEqual(2, len(module_imports))

Example #4

    def test_parser_identifies_import_from(self):
        source = '''
import os
import itertools
from os import path
from gooey.gooey_decorator import Gooey
    '''
        import _ast
        nodes = ast.parse(source)
        from_imports = source_parser.get_nodes_by_instance_type(
            nodes, _ast.ImportFrom)
        self.assertEqual(2, len(from_imports))