def test_univariate(): assert diop_solve((x - 1)*(x - 2)**2) == set([(Integer(1),), (Integer(2),)]) assert diop_solve((x - 1)*(x - 2)) == set([(Integer(1),), (Integer(2),)])