Merge pull request #51 from JuliaAlgebra/bl/variable_union_type

Implement variable_union_type

Author: blegat

Project.toml

@@ -11,6 +11,6 @@ Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
-MultivariatePolynomials = "~0.3"
+MultivariatePolynomials = "~0.3.2"
 Reexport = ">= 0.2"
 julia = "1"

src/DynamicPolynomials.jl

@@ -13,6 +13,7 @@ include("monovec.jl")
 include("poly.jl")
 const TermPoly{C, T} = Union{Term{C, T}, Polynomial{C, T}}
 const PolyType{C} = Union{Polynomial{C}, Term{C}, Monomial{C}, PolyVar{C}}
+MP.variable_union_type(::Union{PolyType{C}, Type{<:PolyType{C}}}) where {C} = PolyVar{C}
 MP.constantmonomial(::Type{<:PolyType{C}}) where {C} = Monomial{C}()
 MP.constantmonomial(p::PolyType) = Monomial(_vars(p), zeros(Int, nvariables(p)))
 MP.monomialtype(::Type{<:PolyType{C}}) where C = Monomial{C}

test/mono.jl

@@ -25,6 +25,18 @@
         @test vars == (x, y, z)
     end
 
+    @testset "variable_union_type" begin
+        @polyvar x
+        @test DynamicPolynomials.MP.variable_union_type(x) == PolyVar{true}
+        @test DynamicPolynomials.MP.variable_union_type(x^2) == PolyVar{true}
+        @test DynamicPolynomials.MP.variable_union_type(2x) == PolyVar{true}
+        @test DynamicPolynomials.MP.variable_union_type(x + 1) == PolyVar{true}
+        @ncpolyvar y
+        @test DynamicPolynomials.MP.variable_union_type(y) == PolyVar{false}
+        @test DynamicPolynomials.MP.variable_union_type(y^2) == PolyVar{false}
+        @test DynamicPolynomials.MP.variable_union_type(2y) == PolyVar{false}
+        @test DynamicPolynomials.MP.variable_union_type(y + 1) == PolyVar{false}
+    end
     @testset "PolyVar" begin
         @test zeroterm(PolyVar{true}) == 0
         @test zero(PolyVar{true}) == 0