More substitution (#40)

Author: blegat

src/arithmetic.jl

@@ -99,6 +99,16 @@ function (p::_AE)(args::MP.AbstractSubstitution...)
     return MP.substitute(MP.Eval(), p, args...)
 end
 
+function (p::_AE)(x::NTuple{N,<:Number}) where {N}
+    return (MP.polynomial(p))(x)
+end
+
+function (p::_AE)(x::AbstractVector{<:Number})
+    return (MP.polynomial(p))(x)
+end
+
+(p::_AE)(x::Number...) = (MP.polynomial(p))(x...)
+
 function MP.differentiate(p::_AE, args...)
     return MP.differentiate(MP.polynomial(p), args...)
 end

test/runtests.jl

@@ -102,6 +102,9 @@ function api_test(B::Type{<:MB.AbstractMonomialIndexed}, degree)
     @test typeof(MB.algebra_element(sum(x))) ==
           MA.promote_operation(MB.algebra_element, typeof(sum(x)))
     @test const_alg_el(x => ones(length(x))) == const_mono
+    @test const_alg_el(ones(length(x))) == const_mono
+    @test const_alg_el(ones(length(x))...) == const_mono
+    @test const_alg_el(tuple(ones(length(x))...)) == const_mono
     @test subs(const_alg_el, x => ones(length(x))) == const_mono
     @test differentiate(const_alg_el, x) == differentiate(const_mono, x)
 end