Add a test for f(x^2).diff(x)