Merge pull request #60 from chriselrod/eachindexarrayoperator

Add eachindex for DiffEqArrayOperator

Author: ChrisRackauckas

src/operators/basic_operators.jl

@@ -108,6 +108,7 @@ Base.eachrow(L::DiffEqArrayOperator) = eachrow(L.A)
 Base.length(L::DiffEqArrayOperator) = length(L.A)
 Base.iterate(L::DiffEqArrayOperator,args...) = iterate(L.A,args...)
 Base.axes(L::DiffEqArrayOperator) = axes(L.A)
+Base.eachindex(L::DiffEqArrayOperator) = eachindex(L.A)
 Base.IndexStyle(::Type{<:DiffEqArrayOperator{T,AType}}) where {T,AType} = Base.IndexStyle(AType)
 Base.copyto!(L::DiffEqArrayOperator, rhs) = (copyto!(L.A, rhs); L)
 Base.copyto!(L::DiffEqArrayOperator, rhs::Base.Broadcast.Broadcasted{<:StaticArrays.StaticArrayStyle}) = (copyto!(L.A, rhs); L)

test/diffeqoperator.jl

@@ -0,0 +1,8 @@
+using SciMLBase
+
+@testset "DiffEqOperator" begin
+    A = rand(10,10);
+    @test eachindex(A) === eachindex(SciMLBase.DiffEqArrayOperator(A))
+    @test eachindex(A') === eachindex(SciMLBase.DiffEqArrayOperator(A'))
+end
+

test/runtests.jl

@@ -21,6 +21,7 @@ if GROUP == "Core" || GROUP == "All"
     @time @safetestset "Existence functions" begin include("existence_functions.jl") end
     @time @safetestset "Integrator interface" begin include("integrator_tests.jl") end
     @time @safetestset "Ensemble functionality" begin include("ensemble_tests.jl") end
+    @time @safetestset "DiffEqOperator tests" begin include("diffeqoperator.jl") end
 end
 
 if !is_APPVEYOR && GROUP == "Downstream"