Add methods in support of PerezHz/TaylorIntegration.jl#192 (#356)

* Add RecursiveArrayTools extension and setindex! dispatch for mixtures

* Add RecursiveArrayTools extension and a Base.setindex! for mixtures

* Fix in setindex!

* Fix test

* Add RAT compat entry (found by Aqua)

Author: PerezHz

Project.toml

@@ -1,7 +1,7 @@
 name = "TaylorSeries"
 uuid = "6aa5eb33-94cf-58f4-a9d0-e4b2c4fc25ea"
 repo = "https://github.com/JuliaDiff/TaylorSeries.jl.git"
-version = "0.17.5"
+version = "0.17.6"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -13,18 +13,21 @@ SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 IntervalArithmetic = "d1acc4aa-44c8-5952-acd4-ba5d80a2a253"
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
+RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
 
 [extensions]
 TaylorSeriesIAExt = "IntervalArithmetic"
 TaylorSeriesJLD2Ext = "JLD2"
 TaylorSeriesSAExt = "StaticArrays"
+TaylorSeriesRATExt = "RecursiveArrayTools"
 
 [compat]
 Aqua = "0.8"
 IntervalArithmetic = "0.15 - 0.20"
 JLD2 = "0.4"
 LinearAlgebra = "<0.0.1, 1"
 Markdown = "<0.0.1, 1"
+RecursiveArrayTools = "2, 3"
 Requires = "0.5.2, 1"
 SparseArrays = "<0.0.1, 1"
 StaticArrays = "1"
@@ -36,9 +39,10 @@ Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 IntervalArithmetic = "d1acc4aa-44c8-5952-acd4-ba5d80a2a253"
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["IntervalArithmetic", "JLD2", "LinearAlgebra", "SparseArrays", "StaticArrays", "Test", "Aqua"]
+test = ["IntervalArithmetic", "JLD2", "LinearAlgebra", "RecursiveArrayTools", "SparseArrays", "StaticArrays", "Test", "Aqua"]

ext/TaylorSeriesRATExt.jl

@@ -0,0 +1,11 @@
+module TaylorSeriesRATExt
+
+using TaylorSeries
+
+isdefined(Base, :get_extension) ? (import RecursiveArrayTools) : (import ..RecursiveArrayTools)
+
+function RecursiveArrayTools.recursivecopy(a::AbstractArray{<:AbstractSeries, N}) where N
+    deepcopy(a)
+end
+
+end

src/auxiliary.jl

@@ -84,6 +84,7 @@ getindex(a::Taylor1{T}, u::StepRange{Int,Int}) where {T<:Number} =
     view(a.coeffs, u[:] .+ 1)
 
 setindex!(a::Taylor1{T}, x::T, n::Int) where {T<:Number} = a.coeffs[n+1] = x
+setindex!(a::Taylor1{T}, x::T, n::Int) where {T<:AbstractSeries} = setindex!(a.coeffs, deepcopy(x), n+1)
 setindex!(a::Taylor1{T}, x::T, u::UnitRange{Int}) where {T<:Number} =
     a.coeffs[u .+ 1] .= x
 function setindex!(a::Taylor1{T}, x::Array{T,1}, u::UnitRange{Int}) where {T<:Number}
@@ -405,4 +406,3 @@ macro isonethread(expr)
         end
     end)
 end
-

test/mixtures.jl

@@ -496,6 +496,12 @@ using Test
     @test two/x == 2/x == 2.0/x
     @test (2one(x))/x == 2/x
 
+    dq = get_variables()
+    x = Taylor1(exp.(dq), 5)
+    x[1] = sin(dq[1]*dq[2])
+    @test x[1] == sin(dq[1]*dq[2])
+    @test x[1] !== sin(dq[1]*dq[2])
+
     @testset "Test Base.float overloads for Taylor1 and TaylorN mixtures" begin
         q = get_variables(Int)
         x1N = Taylor1(q)

test/rat.jl

@@ -0,0 +1,15 @@
+# This file is part of TaylorSeries.jl, MIT licensed
+#
+
+using TaylorSeries, RecursiveArrayTools
+
+using Test
+
+@testset "Tests TaylorSeries RecursiveArrayTools extension" begin
+    dq = get_variables()
+    x = Taylor1([0.9+2dq[1],-1.1dq[1], 0.7dq[2], 0.5dq[1]-0.45dq[2],0.9dq[1]])
+    xx = [x,x]
+    yy = recursivecopy(xx)
+    @test yy == xx # yy and xx are equal...
+    @test yy !== xx # ...but they're not the same object in memory
+end

test/runtests.jl

@@ -12,7 +12,8 @@ testfiles = (
     "identities_Euler.jl",
     "fateman40.jl",
     "staticarrays.jl",
-    "jld2.jl"
+    "jld2.jl",
+    "rat.jl"
     )
 
 for file in testfiles