fix tests

Author: dlfivefifty

src/rect.jl

@@ -40,13 +40,13 @@ end
 function jacobimatrix(::Val{1}, P::RectPolynomial)
     A,B = P.args
     X = jacobimatrix(A)
-    P * KronTrav(Eye{eltype(X)}(∞), X)
+    KronTrav(Eye{eltype(X)}(∞), X)
 end
 # Actually Jyᵀ
 function jacobimatrix(::Val{2}, P::RectPolynomial)
     A,B = P.args
     Y = jacobimatrix(B)
-    P * KronTrav(Y, Eye{eltype(Y)}(∞))
+    KronTrav(Y, Eye{eltype(Y)}(∞))
 end
 @simplify function *(Dx::PartialDerivative{1}, P::RectPolynomial)
     A,B = P.args

test/test_rect.jl

@@ -51,14 +51,15 @@ using ContinuumArrays: plotgridvalues
         Y = jacobimatrix(Val{2}(), TU)
         𝐱 = axes(TU, 1)
         x, y = first.(𝐱), last.(𝐱)
-        @test_broken TU \ (x .* TU) # Should create X, but it fails
-        @test_broken TU \ (y .* TU) # Should create Y, but it fails
+        N = 10
+        KR = Block.(1:N)
+        @test (TU \ (x .* TU))[KR,KR] == X[KR,KR]
+        @test (TU \ (y .* TU))[KR,KR] == Y[KR,KR]
         f = expand(TU, splat((x,y) -> exp(x*cos(y-0.1))))
         g = expand(TU, splat((x,y) -> x*exp(x*cos(y-0.1))))
         h = expand(TU, splat((x,y) -> y*exp(x*cos(y-0.1))))
-        N = 10
-        @test (TU \ (X * (TU \ f)))[Block.(1:N)] ≈ (TU \ g)[Block.(1:N)]
-        @test (TU \ (Y * (TU \ f)))[Block.(1:N)] ≈ (TU \ h)[Block.(1:N)]
+        @test (X * (TU \ f))[KR] ≈ (TU \ g)[KR]
+        @test (Y * (TU \ f))[KR] ≈ (TU \ h)[KR]
     end
 
     @testset "Conversion" begin