def test_get_callees(self):
program = """
fn f() -> u32 { u32:42 }
fn main() -> u32 { f() }
"""
m, node_to_type = self._get_module(program)
self.assertEqual(
((m.get_function('f'), m, node_to_type, ()), ),
extract_conversion_order.get_callees(m.get_function('main'),
m,
node_to_type,
imports={},
bindings=()))
def test_get_callees(self):
program = """
fn f() -> u32 { u32:42 }
fn main() -> u32 { f() }
"""
m, type_info = self._get_module(program)
callee = extract_conversion_order.Callee(m.get_function('f'), m,
type_info, SymbolicBindings())
self.assertEqual(
(callee, ),
extract_conversion_order.get_callees(m.get_function('main'),
m,
type_info,
imports={},
bindings=SymbolicBindings()))