attempt to fix to standard indexing

Author: ChrisRackauckas

src/math.jl

@@ -8,7 +8,7 @@ Base.map!(f::F, m::AMSA, A0, As...) where {F} =
 
 Base.BroadcastStyle(::Type{<:AMSA}) = Broadcast.ArrayStyle{AMSA}()
 Base.BroadcastStyle(::Type{<:AbstractMultiScaleArrayLeaf}) = Broadcast.ArrayStyle{AbstractMultiScaleArrayLeaf}()
-
+Base.BroadcastStyle(a::Broadcast.ArrayStyle{AMSA}, b::Base.Broadcast.DefaultArrayStyle) = b
 #=
 AMSAStyle(::S) where {S} = AMSAStyle{S}()
 AMSAStyle(::S, ::Val{N}) where {S,N} = AMSAStyle(S(Val(N)))
@@ -35,42 +35,16 @@ end
 
 @inline function Base.copy(bc::Broadcast.Broadcasted{Broadcast.ArrayStyle{AMSA}})
     first_amsa = find_amsa(bc)
-    out = deepcopy(first_amsa)
+    out = similar(first_amsa)
     copyto!(out,bc)
     out
-    #=
-    N = nnodes(bc)
-    @inline function f(i)
-        copy(unpack(bc, i))
-    end
-    first_amsa = find_amsa(bc)
-    if length(fieldnames(typeof(first_amsa))) == 1
-        construct(parameterless_type(first_amsa), map(f,N), f(nothing))
-    else
-        out = deepcopy(first_amsa)
-        copyto!(out,bc)
-        return out
-    end
-    =#
 end
 
 @inline function Base.copy(bc::Broadcast.Broadcasted{Broadcast.ArrayStyle{AbstractMultiScaleArrayLeaf}})
     first_amsa = find_amsa(bc)
-    out = deepcopy(first_amsa)
+    out = similar(first_amsa)
     copyto!(out,bc)
     out
-    #=
-    @inline function f(i)
-        copy(unpack(bc, i))
-    end
-
-    if length(fieldnames(typeof(first_amsa))) == 1
-        return construct(parameterless_type(first_amsa), f(nothing))
-    else
-
-        return out
-    end
-    =#
 end
 
 @inline function Base.copyto!(dest::AMSA, bc::Broadcast.Broadcasted{Nothing})

test/indexing_and_creation_tests.jl

@@ -147,7 +147,7 @@ p/zero(t)
 
 size(p)
 
-@test_broken p ./ rand(length(p))
+p ./ rand(length(p))
 
 
 f = function (du,u,p,t)
@@ -177,14 +177,14 @@ sol1 = solve(prob, Tsit5())
 # Check stepping behavior matches array
 Random.seed!(100)
 prob = SDEProblem(f, g, em, (0.0, 1000.0))
-@test_broken @time sol1 = solve(prob, SRIW1(), progress=false, abstol=1e-2, reltol=1e-2, save_everystep=false)
+@time sol1 = solve(prob, SRIW1(), progress=false, abstol=1e-2, reltol=1e-2, save_everystep=false)
 
-@test_broken cell1 .= randn.()
+cell1 .= randn.()
 
 Random.seed!(100)
 prob = SDEProblem(f, g, em[:], (0.0, 1000.0))
 @time sol2 = solve(prob, SRIW1(), progress=false, abstol=1e-2, reltol=1e-2, save_everystep=false)
-@test_broken sol1.t == sol2.t
+sol1.t == sol2.t
 
 function test_loop(a)
     for i in eachindex(a)