def test_unbound_reference_to_locals_and_arguments(self):
_, result = bootstrap_function(function_lit(
object_type({"foo": int_type()}), infer_all(),
object_type({"bar": int_type()}),
object_template_op({"bar": literal_op(3)}),
return_op(
addition_op(unbound_dereference("foo"),
unbound_dereference("bar")))),
argument=PythonObject({"foo": 39}))
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, 42)
def type_conditional_converter(expression):
is_conditional = expression.opcode == "conditional"
if not is_conditional:
return expression
condition_is_type_check = expression.condition.opcode == "is"
if not condition_is_type_check:
return expression
lvalue_of_condition_is_dereference = expression.condition.expression.opcode == "unbound_dereference"
if not lvalue_of_condition_is_dereference:
return expression
shadow_name = expression.condition.expression.reference
new_match = match_op(expression.condition.expression, [
prepared_function(
expression.condition.type,
invoke_op(prepared_function(
object_type({shadow_name: expression.condition.type}),
expression.when_true),
argument_expression=object_template_op(
{shadow_name: dereference("argument")}))),
prepared_function(inferred_type(), expression.when_false)
])
get_manager(new_match).add_composite_type(DEFAULT_READONLY_COMPOSITE_TYPE)
return new_match
def test_interesting(self):
func = function_lit(
any_type(), infer_all(),
match_op(dereference("argument"), [
prepared_function(
object_type({"foo": int_type()}),
return_op(
addition_op(dereference("argument.foo"),
literal_op(3)))),
prepared_function(any_type(), return_op(literal_op("invalid")))
]))
_, result = bootstrap_function(func,
argument=PythonObject(
{"foo": 39},
bind=UniversalObjectType(
{"foo": IntegerType()})))
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, 42)
_, result = bootstrap_function(func,
argument=PythonObject({"foo": "hello"}))
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, "invalid")
def test_return_with_dereference4(self):
_, result = bootstrap_function(function_lit(
int_type(),
build_break_types(
object_type({
"foo": any_type(),
"bar": any_type()
})),
comma_op(
return_op(
object_template_op({
"foo":
literal_op(42),
"bar":
dereference_op(context_op(), literal_op("argument"),
True)
})))),
argument=42,
check_safe_exit=True)
self.assertEquals(result.caught_break_mode, "return")
self.assertTrue(isinstance(result.value, Universal))
get_manager(
result.value).add_composite_type(DEFAULT_READONLY_COMPOSITE_TYPE)
self.assertEquals(result.value._get("foo"), 42)
self.assertEquals(result.value._get("bar"), 42)
def test_nested_return(self):
_, result = bootstrap_function(function_lit(
no_value_type(),
build_break_types(
object_type({"foo": object_type({"bar": int_type()})})),
comma_op(
return_op(
object_template_op(
{"foo": object_template_op({"bar":
literal_op(42)})})))),
check_safe_exit=True)
self.assertEquals(result.caught_break_mode, "return")
self.assertTrue(isinstance(result.value, Universal))
get_manager(
result.value).add_composite_type(DEFAULT_READONLY_COMPOSITE_TYPE)
self.assertEquals(result.value._get("foo")._get("bar"), 42)
def test_catch_real_exception(self):
# Function safely handles an internal exception
func = function_lit(
try_catch_op(
dereference_op(context_op(), literal_op("foo"), True),
prepared_function(
object_type({
"type": const_string_type(),
"message": const_string_type(),
}), return_op(dereference("argument.message"))), nop()))
_, result = bootstrap_function(func)
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, "DereferenceOp: invalid_dereference")
def visitToListDestructuring(self, ctx):
lvalues = [
self.visit(l) for l in ctx.assignmentOrInitializationLvalue()
]
rvalue = self.visit(ctx.expression())
code_block = ctx.codeBlock()
if code_block:
code_block = self.visit(code_block)
assignments = [(i, n) for i, n in enumerate(lvalues)
if isinstance(n, basestring)]
initializers = [(i, n) for i, n in enumerate(lvalues)
if isinstance(n, tuple)]
local_variable_types = {}
local_variable_initializers = {}
for index, (type, name) in initializers:
local_variable_types[name] = type
local_variable_initializers[name] = dereference(
"outer.local._temp", index)
code_block = CodeBlockBuilder(
local_variable_type=object_type(local_variable_types),
local_initializer=object_template_op(
local_variable_initializers)).chain(code_block,
get_debug_info(ctx))
code_block = CodeBlockBuilder(
local_variable_type=object_type(
{"_temp": list_type([inferred_type()] * len(lvalues), None)}),
local_initializer=object_template_op({"_temp": rvalue}),
code_expressions=[
unbound_assignment(name, dereference("local._temp", index))
for index, name in assignments
]).chain(code_block, get_debug_info(ctx))
return code_block
def visitArgumentDestructurings(self, ctx):
initializers = [self.visit(l) for l in ctx.argumentDestructuring()]
argument_types = [None] * len(initializers)
local_variable_types = {}
local_variable_initializers = {}
for index, (type, name) in enumerate(initializers):
argument_types[index] = type
local_variable_types[name] = type
local_variable_initializers[name] = dereference("argument", index)
return CodeBlockBuilder(
argument_type_expression=list_type(argument_types, None),
local_variable_type=object_type(local_variable_types),
local_initializer=object_template_op(local_variable_initializers))
def test_misc1(self):
_, result = bootstrap_function(
function_lit(
no_value_type(), infer_all(),
object_type({
"foo": int_type(),
"bar": int_type()
}),
object_template_op({
"foo": literal_op(39),
"bar": literal_op(3)
}),
return_op(
addition_op(dereference("local.foo"),
dereference("local.bar")))))
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, 42)
def visitForGeneratorLoop(self, ctx):
iterator_name = ctx.SYMBOL().getText()
generator_expression = self.visit(ctx.expression())
loop_code = self.visit(ctx.codeBlock())
loop_code = CodeBlockBuilder(argument_type_expression=object_type(
{iterator_name: inferred_type()})).chain(loop_code,
get_debug_info(ctx))
loop_code = loop_code.create("second-class-function",
get_debug_info(ctx))
# get_manager(loop_code).add_composite_type(READONLY_DEFAULT_OBJECT_TYPE)
return transform_op(
"break", "value",
invoke_op(
local_function(
object_template_op({"callback": generator_expression}),
loop_op(
invoke_op(
local_function(
transform(
("yield", "value"), ("value", "end"),
reset_op(
dereference("outer.local.callback",
**get_debug_info(ctx)),
nop(), **get_debug_info(ctx)),
**get_debug_info(ctx)),
comma_op(
assignment_op(
dereference("outer.local"),
literal_op("callback"),
dereference("local.continuation"),
**get_debug_info(ctx)),
invoke_op(
prepare_function_lit(loop_code),
object_template_op(
{
iterator_name:
dereference("local.value")
}, **get_debug_info(ctx)),
**get_debug_info(ctx))),
**get_debug_info(ctx)), **get_debug_info(ctx)),
**get_debug_info(ctx)), **get_debug_info(ctx))))
def test_return_with_dereference5(self):
with self.assertRaises(Exception):
bootstrap_function(function_lit(
int_type(),
build_break_types(
object_type({
"foo": any_type(),
"bar": unit_type(42)
})),
return_op(
object_template_op({
"foo":
literal_op(42),
"bar":
dereference_op(context_op(), literal_op("argument"),
True)
}))),
argument=42,
check_safe_exit=True)
def visitLocalVariableDeclaration(self, ctx):
type = self.visit(ctx.expression())
remaining_code = ctx.codeBlock()
if remaining_code:
remaining_code = self.visit(remaining_code)
for symbol_initialization in reversed(ctx.symbolInitialization()):
name, initial_value = self.visit(symbol_initialization)
new_code_block = CodeBlockBuilder(
local_variable_type=object_type({name: type}),
local_initializer=object_template_op({name: initial_value}))
if remaining_code:
new_code_block = new_code_block.chain(remaining_code,
get_debug_info(ctx))
remaining_code = new_code_block
return new_code_block
def test_silly_tostring_casing(self):
func = function_lit(
any_type(),
try_catch_op(
return_op(
match_op(dereference("argument"), [
prepared_function(unit_type(1), literal_op("one")),
prepared_function(unit_type(2), literal_op("two")),
prepared_function(
int_type(),
throw_op(
object_template_op(
{"type": literal_op("UnknownInt")}))),
prepared_function(
any_type(),
throw_op(
object_template_op(
{"type": literal_op("TypeError")})))
])),
prepared_function(
object_type({"type": unit_type("UnknownInt")}),
return_op(literal_op("unknown")))))
_, result = bootstrap_function(func, argument=1, check_safe_exit=False)
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, "one")
_, result = bootstrap_function(func, argument=2, check_safe_exit=False)
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, "two")
_, result = bootstrap_function(func, argument=3, check_safe_exit=False)
self.assertEquals(result.caught_break_mode, "return")
self.assertEquals(result.value, "unknown")
_, result = bootstrap_function(func,
argument="hello",
check_safe_exit=False)
self.assertEquals(result.caught_break_mode, "exception")
self.assertIsInstance(result.value, Universal)
self.assertEquals(result.value._get("type"), "TypeError")
def visitForListLoop(self, ctx):
iterator_name = ctx.SYMBOL().getText()
composite_expression = self.visit(ctx.expression())
loop_code = self.visit(ctx.codeBlock())
loop_code = CodeBlockBuilder(argument_type_expression=object_type(
{iterator_name: inferred_type()})).chain(loop_code,
get_debug_info(ctx))
loop_code = loop_code.create("second-class-function",
get_debug_info(ctx))
return map_op(
composite_expression,
prepared_function(
inferred_type(),
invoke_op(
prepare_function_lit(loop_code, **get_debug_info(ctx)),
object_template_op(
{
iterator_name:
dereference("argument", **get_debug_info(ctx))
}, **get_debug_info(ctx)), **get_debug_info(ctx)),
**get_debug_info(ctx)))
def create(self, output_mode, debug_info):
if output_mode not in ("first-class-function", "second-class-function",
"expression"):
raise FatalError()
code_expressions = default(self.code_expressions, MISSING, [])
for c in code_expressions:
if not is_opcode(c):
raise FatalError()
if not self.requires_function() and output_mode == "expression":
return combine_opcodes(code_expressions)
if self.argument_type_expression is not MISSING:
argument_type = self.argument_type_expression
else:
argument_type = no_value_type()
if self.local_variable_type is not MISSING:
local_type = self.local_variable_type
else:
local_type = object_type(
{}) # For future python local variables...
if self.local_initializer is not MISSING:
local_initializer = self.local_initializer
else:
local_initializer = object_template_op({})
if self.breaks_types is not MISSING:
break_types = self.breaks_types
else:
break_types = infer_all()
if self.extra_statics is not MISSING:
extra_statics = self.extra_statics
else:
extra_statics = {}
if output_mode == "first-class-function":
# A function created by the user, which mangles returns as expected
code = transform_op("return", "value",
combine_opcodes(code_expressions),
**debug_info)
return function_lit(extra_statics, argument_type, break_types,
local_type, local_initializer, code,
**debug_info)
if output_mode == "second-class-function":
# A function created by the environment, which leaves returns unmangled
code = combine_opcodes(code_expressions)
return function_lit(extra_statics, argument_type, break_types,
local_type, local_initializer, code,
**debug_info)
elif output_mode == "expression":
return invoke_op(
prepare_function_lit(
function_lit(extra_statics, argument_type, break_types,
local_type, local_initializer,
combine_opcodes(code_expressions),
**debug_info), **debug_info), **debug_info)
def visitObjectType(self, ctx):
result = {}
for pair in ctx.objectTypePropertyPair():
name, type = self.visit(pair)
result[name] = type
return object_type(result)
