def known_types_for_function(source_code, name): real_source_code = dedent(source_code) compiled = analyzer.parse_code(real_source_code) t = analyzer.SymbolTableTypeInfer(compiled) t.bind_types() known = t.known_types(scope_name=name) return known
def known_types_for_module(source_code): real_source_code = dedent(source_code) compiled = analyzer.parse_code(real_source_code) t = analyzer.SymbolTableTypeInfer(compiled) t.bind_types() known = t.known_types() return known