def test_extract_attr(): code = ''' import sys sys.stdin.write('hello, world!\\n') ''' module_node = parse_source(code) module_accessors = list(get_module_accessors(module_node)) # assert 'sys.stdin' in module_accessors assert 'sys.stdin.write' in module_accessors
def main(argv): try: filename = argv[1] except IndexError: print 'usage: %s [filename]' % argv[0] return 1 # with open(filename) as f: module_node = parse_source(f.read()) import_names = list(get_import_names(module_node)) accessors = list(get_module_accessors(module_node)) print import_names print accessors
def test_extract_attr_with_from_and_as(): code = ''' from sys import stdin as mystdin mystdin.write('hello, world!\\n') from os.path import dirname as mydirname print mydirname(__file__) ''' module_node = parse_source(code) module_accessors = list(get_module_accessors(module_node)) # assert 'sys.stdin.write' in module_accessors assert 'os.path.dirname' in module_accessors