TST: fix PolynomialFeatures test error