def test_restart_into_local_initialization(self): context = PythonObject({}) frame_manager = FrameManager() func = prepare( function_lit( no_value_type(), build_break_types(any_type(), yield_types={ "out": any_type(), "in": int_type() }), int_type(), shift_op(literal_op("hello"), int_type()), return_op( dereference_op(context_op(), literal_op("local"), True))), context, frame_manager).close(None) def start(): func.invoke(NO_VALUE, frame_manager) with frame_manager.capture("yield") as yielder: start() self.assertEquals(yielder.value, "hello") def restart(): yielder_restart_continuation = yielder.create_continuation( start, func.get_type().break_types) yielder_restart_continuation.invoke(32, frame_manager) with frame_manager.capture("return") as returner: restart() self.assertEquals(returner.value, 32)
def test_immediate_return(self): _, result = bootstrap_function( function_lit(no_value_type(), build_break_types(int_type()), loop_op(return_op(literal_op(42))))) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_count_then_return(self): _, result = bootstrap_function( function_lit( no_value_type(), build_break_types(int_type()), int_type(), literal_op(0), loop_op( comma_op( assignment_op( context_op(), literal_op("local"), addition_op( dereference_op(context_op(), literal_op("local"), True), literal_op(1))), condition_op( equality_op( dereference_op(context_op(), literal_op("local"), True), literal_op(42)), return_op( dereference_op(context_op(), literal_op("local"))), nop()))))) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_basic_inferrence(self): _, result = bootstrap_function( function_lit(no_value_type(), build_break_types(inferred_type()), return_op(literal_op(42)))) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_single_restart(self): context = PythonObject({}) frame_manager = FrameManager() func = prepare( function_lit( no_value_type(), build_break_types(any_type(), yield_types={ "out": any_type(), "in": int_type() }), return_op( addition_op(shift_op(literal_op("hello"), int_type()), literal_op(40)))), context, frame_manager).close(None) def first(): func.invoke(NO_VALUE, frame_manager) with frame_manager.capture("yield") as yielder: first() self.assertEquals(yielder.value, "hello") def second(): yielder.create_continuation(first, {}).invoke(2, frame_manager) with frame_manager.capture("return") as returner: second() self.assertEquals(returner.value, 42)
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_comma(self): _, result = bootstrap_function(function_lit( no_value_type(), build_break_types(int_type()), return_op(comma_op(literal_op(5), literal_op(8), literal_op(42)))), check_safe_exit=True) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_addition(self): func = function_lit( no_value_type(), build_break_types(int_type()), return_op(addition_op(literal_op(40), literal_op(2)))) _, result = bootstrap_function(func) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_basic_function(self): func = function_lit(no_value_type(), build_break_types(value_type=int_type()), literal_op(42)) _, result = bootstrap_function(func) self.assertEquals(result.caught_break_mode, "value") self.assertEquals(result.value, 42)
def test_basic(self): _, result = bootstrap_function( function_lit( no_value_type(), build_break_types(int_type()), return_op( invoke_op( close_op( static_op( prepare_op( literal_op( function_lit( no_value_type(), build_break_types(int_type()), return_op(literal_op(42)))))), context_op()))))) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_basic_false(self): _, result = bootstrap_function( function_lit( no_value_type(), build_break_types(int_type()), return_op( condition_op(literal_op(False), literal_op(34), literal_op(53))))) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 53)
def test_simple_return_argument(self): func = function_lit( int_type(), build_break_types(int_type()), return_op( dereference_op(context_op(), literal_op("argument"), True))) _, result = bootstrap_function(func, argument=42) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_initialization(self): _, result = bootstrap_function(function_lit( no_value_type(), build_break_types(int_type()), int_type(), literal_op(42), comma_op( return_op( dereference_op(context_op(), literal_op("local"), True)))), check_safe_exit=True) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_return(self): _, result = bootstrap_function(function_lit( no_value_type(), build_break_types(object_type({"foo": int_type()})), comma_op(return_op(object_template_op({"foo": 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"), 42)
def test_assignment_from_argument(self): _, result = bootstrap_function(function_lit( int_type(), build_break_types(int_type()), int_type(), literal_op(0), comma_op( assignment_op( context_op(), literal_op("local"), dereference_op(context_op(), literal_op("argument"), True)), return_op( dereference_op(context_op(), literal_op("local"), True)))), argument=43, check_safe_exit=True) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 43)
def test_restart_comma(self): context = PythonObject({}) frame_manager = FrameManager() func = prepare( function_lit( no_value_type(), build_break_types(int_type(), yield_types={ "out": any_type(), "in": int_type() }), return_op( comma_op(literal_op(5), shift_op(literal_op("first"), int_type()), shift_op(literal_op("second"), int_type())))), context, frame_manager).close(None) def first(): func.invoke(NO_VALUE, frame_manager) with frame_manager.capture("yield") as first_yielder: first() self.assertEquals(first_yielder.value, "first") def second(): first_yield_restart_continuation = first_yielder.create_continuation( first, func.get_type().break_types) first_yield_restart_continuation.invoke(4, frame_manager) with frame_manager.capture("yield") as second_yielder: second() self.assertEquals(second_yielder.value, "second") def third(): second_yield_restart_continuation = second_yielder.create_continuation( second, func.get_type().break_types) second_yield_restart_continuation.invoke(42, frame_manager) with frame_manager.capture("return") as returner: third() self.assertEquals(returner.value, 42)
def test_restart_into_local_initialization_and_code(self): context = PythonObject({}) frame_manager = FrameManager() func = prepare( function_lit( no_value_type(), build_break_types(any_type(), yield_types={ "out": any_type(), "in": int_type() }), int_type(), shift_op(literal_op("first"), int_type()), return_op( addition_op( dereference_op(context_op(), literal_op("local"), True), shift_op(literal_op("second"), int_type())))), context, frame_manager).close(None) def first(): func.invoke(NO_VALUE, frame_manager) with frame_manager.capture("yield") as first_yielder: first() self.assertEquals(first_yielder.value, "first") def second(): first_restart_continuation = first_yielder.create_continuation( first, func.get_type().break_types) first_restart_continuation.invoke(40, frame_manager) with frame_manager.capture("yield") as second_yielder: second() self.assertEquals(second_yielder.value, "second") with frame_manager.capture("return") as returner: second_restart_continuation = second_yielder.create_continuation( first, func.get_type().break_types) second_restart_continuation.invoke(2, frame_manager) self.assertEquals(returner.value, 42)
def test_basic_with_inferred_types(self): _, result = bootstrap_function( function_lit( no_value_type(), infer_all(), function_type(no_value_type(), build_break_types(int_type())), close_op( static_op( prepare_op( literal_op( function_lit(no_value_type(), infer_all(), return_op(literal_op(42)))))), context_op()), return_op( invoke_op( dereference_op(context_op(), literal_op("local"), True))))) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
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 test_add_locals(self): func = function_lit( no_value_type(), build_break_types(int_type()), return_op( addition_op( dereference_op( dereference_op( dereference_op(context_op(), literal_op("outer"), True), literal_op("local"), True), literal_op(0), True), dereference_op( dereference_op( dereference_op(context_op(), literal_op("outer"), True), literal_op("local"), True), literal_op(1), True)))) context = PythonObject( { "local": PythonList([39, 3]), "_types": PythonObject({ "local": UniversalTupleType([IntegerType(), IntegerType()]) }) }, bind=UniversalObjectType({ "local": UniversalTupleType([IntegerType(), IntegerType()]), "_types": DEFAULT_READONLY_COMPOSITE_TYPE })) _, result = bootstrap_function(func, outer_context=context) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_repeated_restart_with_outer_return_handling(self): context = PythonObject({}) frame_manager = FrameManager() func = prepare( function_lit( no_value_type(), build_break_types(int_type(), yield_types={ "out": any_type(), "in": int_type() }), return_op( addition_op(shift_op(literal_op("first"), int_type()), shift_op(literal_op("second"), int_type())))), context, frame_manager).close(None) with frame_manager.capture("return") as return_capturer: def first(): func.invoke(NO_VALUE, frame_manager) with frame_manager.capture("yield") as first_yielder: first() self.assertEquals(first_yielder.value, "first") def second(): first_yielder.create_continuation(first, {}).invoke( 39, frame_manager) with frame_manager.capture("yield") as second_yielder: second() self.assertEquals(second_yielder.value, "second") second_yielder.create_continuation(second, {}).invoke(3, frame_manager) self.assertEquals(return_capturer.value, 42)
def test_repeated_restart_while_using_restart_values(self): context = PythonObject({}) frame_manager = FrameManager() func = prepare( function_lit( no_value_type(), build_break_types(any_type(), yield_types={ "out": any_type(), "in": int_type() }), return_op( addition_op(shift_op(literal_op(30), int_type()), shift_op(literal_op(10), int_type())))), context, frame_manager).close(None) def first(): func.invoke(NO_VALUE, frame_manager) with frame_manager.capture("yield") as first_yielder: first() def second(): first_yielder.create_continuation(first, {}).invoke( first_yielder.value + 1, frame_manager) with frame_manager.capture("yield") as second_yielder: second() def third(): second_yielder.create_continuation(second, {}).invoke( second_yielder.value + 1, frame_manager) with frame_manager.capture("return") as returner: third() self.assertEquals(returner.value, 42)
def test_return_variable(self): func = function_lit( no_value_type(), build_break_types(int_type()), return_op( dereference_op( dereference_op(context_op(), literal_op("outer"), True), literal_op("local"), True))) context = PythonObject( { "local": 42, "_types": PythonObject({"local": IntegerType()}) }, bind=UniversalObjectType( { "local": IntegerType(), "_types": DEFAULT_READONLY_COMPOSITE_TYPE }, wildcard_type=RICH_READONLY_TYPE)) _, result = bootstrap_function(func, outer_context=context) self.assertEquals(result.caught_break_mode, "return") self.assertEquals(result.value, 42)
def test_without_infer_exception_fails(self): with self.assertRaises(Exception): bootstrap_function( function_lit(no_value_type(), build_break_types(int_type()), addition_op(literal_op("hello"), literal_op(5))))