Exemplo n.º 1
0
 def test_statements_in_module_body_are_transformed(self):
     module_node = nodes.module([nodes.expression_statement(nodes.ref("value"))], is_executable=True)
     module_type = types.module("blah", [])
     _assert_transform(
         module_node,
         cc.module([cc.expression_statement(cc.ref("value"))], is_executable=True, exported_names=[]),
         type_lookup=[(module_node, module_type)]
     )
Exemplo n.º 2
0
def builtin_modules_are_typed():
    cgi_module = BuiltinModule("cgi", types.module("cgi", [
        types.attr("escape", types.none_type),
    ]))
    node = nodes.Import([nodes.import_alias("cgi", None)])
    
    context = update_blank_context(
        node,
        module_resolver=FakeModuleResolver({("cgi",): cgi_module}),
        module_types=FakeModuleTypes({}),
    )
    
    assert_equal(types.none_type, context.lookup_name("cgi").attrs.type_of("escape"))
Exemplo n.º 3
0
 def test_module_exports_are_set_directly_on_module(self):
     module_node = nodes.module(
         [nodes.assign([nodes.ref("value")], nodes.none())],
         is_executable=False
     )
     module_type = types.module("blah", [types.attr("value", types.none_type)])
     _assert_transform(
         module_node,
         cc.module(
             [
                 cc.declare("value"),
                 cc.assign(cc.ref("value"), cc.none)
             ],
             is_executable=False,
             exported_names=["value"],
         ),
         type_lookup=[(module_node, module_type)]
     )
Exemplo n.º 4
0
 def value_in_builtin_module_can_be_resolved(self):
     # import cgi
     cgi_module = BuiltinModule(
         "cgi",
         types.module("cgi", [types.attr("escape", types.none_type)])
     )
     
     module_resolver = _module_resolver(
         _create_module("root/main.py", is_executable=True),
         builtin_modules={"cgi": cgi_module}
     )
     resolved_import = module_resolver.resolve_import_value(
         ["cgi"],
         "escape",
     )
     
     assert_is(cgi_module, resolved_import.module)
     assert_equal("escape", resolved_import.attr_name)
     assert_equal(["cgi"], resolved_import.module_name)
Exemplo n.º 5
0
def module(name, attrs=None):
    if attrs is None:
        attrs = []
    return types.module(name, attrs)