Fix polynomial for PolynomialVectorBasis

Author: blegat

Project.toml

@@ -1,7 +1,7 @@
 name = "MultivariateBases"
 uuid = "be282fd4-ad43-11e9-1d11-8bd9d7e43378"
 repo = "https://github.com/JuliaAlgebra/MultivariateBases.jl.git"
-version = "0.1.1"
+version = "0.1.2"
 
 [deps]
 MultivariatePolynomials = "102ac46a-7ee4-5c85-9060-abc95bfdeaa3"

src/fixed.jl

@@ -15,17 +15,17 @@ function MP.polynomialtype(basis::AbstractPolynomialVectorBasis{PT}, T::Type) wh
     return MP.polynomialtype(PT, V)
 end
 function MP.polynomial(f::Function, basis::AbstractPolynomialVectorBasis)
-    return mapreduce(ip -> f(ip[1]) * ip[2], MA.add!, enumerate(basis.polynomials))
+    return MP.polynomial(mapreduce(
+        ip -> f(ip[1]) * ip[2], MA.add!, enumerate(basis.polynomials)))
 end
 
 function MP.polynomial(Q::AbstractMatrix, basis::AbstractPolynomialVectorBasis,
                        T::Type)
     n = length(basis)
     @assert size(Q) == (n, n)
-    return mapreduce(row -> basis.polynomials[row] *
+    return MP.polynomial(mapreduce(row -> basis.polynomials[row] *
         mapreduce(col -> Q[row, col] * basis.polynomials[col], MA.add!, 1:n),
-        MA.add!, 1:n)
-    return mapreduce()
+        MA.add!, 1:n), T)
 end
 
 """

test/fixed.jl

@@ -27,4 +27,7 @@ end
     basis = FixedPolynomialBasis([x])
     @test polynomialtype(basis, Int) == polynomialtype(x, Int)
     @test polynomial(i -> 5, basis) == 5x
+    @test typeof(polynomial(i -> 5, basis)) == polynomialtype(basis, Int)
+    @test typeof(polynomial(ones(Int, 1, 1), basis, Int)) <: AbstractPolynomial{Int}
+    @test typeof(polynomial(ones(Int, 1, 1), basis, Float64)) <: AbstractPolynomial{Float64}
 end