def test_variables(): u = Variable('u') c = Constant('c') strong_form = StrongForm(c*u) assert(strong_form.variables() == set([u]))