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))
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))
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))
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))