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'})
