def test_variables(self): """ Testing that the variables of methods is returned correctly. """ method = Method(node('x'), node('y')) variables = method.variables() assert_equal(variables, {'x', 'y'})
def test_variables_with_internal_nodes(self): """ Testing that variables is retured correctly, with internal nodes """ from fbml import buildin method = Method( node(buildin.lt, {'a': node('number'), 'b': node('const')}), node(buildin.mul, {'a': node('number'), 'b': node('const')}) ) variables = method.variables() assert_equal(variables, {'number', 'const'})