def test_application(self):
self._registry.push_new_scope({'times2': ('var_times2_1', True)})
v = Variable('times2')
l = Literal('Int', 123)
a = Application(v, [l])
a_id = a.add_to_rules(self._rules, self._registry)
v_id = self._registry.get_id_for(v)
l_id = self._registry.get_id_for(l)
result = self._rules.infer()
self.assertEqual('Int', result.get_type_by_id(l_id))
self.assertEqual(None, result.get_type_by_id(a_id))
def test_application(self):
scoped_id = 'var_times2_1'
self._registry.push_new_scope({'times2': (scoped_id, True)})
v = Variable('times2')
l = Literal('Int', 123)
a = Application(v, [l])
a_id = a.add_to_rules(self._rules, self._registry)
v_id = self._registry.get_id_for(v)
l_id = self._registry.get_id_for(l)
self.assertIn((v_id, scoped_id), self._rules.instance_of_calls)
self.assertIn((v_id, ('Fn_1', l_id, a_id)), self._rules.specify_calls)
def test_polymorphism(self):
''' ML code:
let id = \\x -> x
in (id id) 123
'''
lm = Lambda(['x'], Variable('x'))
app1 = Application(Variable('id'), [Variable('id')])
app2 = Application(app1, [Literal('Int', 123)])
lt = Let([('id', lm)], app2)
lt_id = lt.add_to_rules(self._rules, self._registry)
result = self._rules.infer()
self.assertEqual('Int', result.get_type_by_id(lt_id))
def test_let_with_lambda(self):
lm = Lambda(['x'], Variable('x'))
var_id = Variable('id')
app = Application(var_id, [Literal('Int', 123)])
lt = Let([('id', lm)], app)
lt_id = lt.add_to_rules(self._rules, self._registry)
app_id = self._registry.get_id_for(app)
self.assertIn((lt_id, app_id), self._rules.equal_calls)
def test_let_with_lambda(self):
''' ML code:
let id = \\x -> x
in id 'foo'
'''
lm = Lambda(['x'], Variable('x'))
var_id = Variable('id')
app = Application(var_id, [Literal('String', 'foo')])
lt = Let([('id', lm)], app)
lt_id = lt.add_to_rules(self._rules, self._registry)
result = self._rules.infer()
self.assertEqual('String', result.get_type_by_id(lt_id))
def test_generic_mutual_recursion(self):
'''
Equivalent ML:
let-rec f x = if True then x else g x
g y = f y
in g
'''
test = Literal('Bool', True)
if_case = Variable('x')
else_case = Application(Variable('g'), [Variable('x')])
if_block = If(test, if_case, else_case)
f_func = Lambda(['x'], if_block)
g_body = Application(Variable('f'), [Variable('y')])
g_func = Lambda(['y'], g_body)
let_body = Variable('f')
let_expr = Let([('f', f_func), ('g', g_func)], let_body)
let_id = let_expr.add_to_rules(self._rules, self._registry)
result = self._rules.infer()
self.assertEqual(('Fn_1', 'a0', 'a0'), result.get_full_type_by_id(let_id))
def test_mutual_recursion(self):
'''
Equivalent ML:
let-rec f = if True then 123 else g
g = f
in f
'''
test = Literal('Bool', True)
if_case = Literal('Int', 123)
else_case = Application(Variable('g'), [])
if_block = If(test, if_case, else_case)
f_func = Lambda([], if_block)
g_body = Application(Variable('f'), [])
g_func = Lambda([], g_body)
let_body = Variable('f')
let_expr = Let([('f', f_func), ('g', g_func)], let_body)
let_id = let_expr.add_to_rules(self._rules, self._registry)
result = self._rules.infer()
self.assertEqual(('Fn_0', 'Int'), result.get_full_type_by_id(let_id))