def test_product_generator_iterator_variables(): x = Variable(Int(), "x") y = Variable(Int(), "x") p = Product((Range(x), "x"), (Range(y), "y")) assert set() == set(p.get_variables()) assert {x, y} == set(p.iterate().get_variables()) assert {x, y} == set(p.generate().get_variables())