test: use updated API for the tests

Author: avik-pal

src/Compiler.jl

@@ -10,6 +10,7 @@ import ..Reactant:
     ConcreteRNumber,
     TracedRArray,
     TracedRNumber,
+    RArray,
     OrderedIdDict,
     make_tracer,
     TracedToConcrete,
@@ -19,10 +20,13 @@ import ..Reactant:
 @inline traced_getfield(@nospecialize(obj), field) = Base.getfield(obj, field)
 @inline traced_getfield(@nospecialize(obj::AbstractArray), field) =
     Base.getindex(obj, field)
+@inline traced_getfield(@nospecialize(obj::RArray), field) = Base.getfield(obj, field)
 
-@inline traced_setfield!(@nospecialize(obj), field, val) = Base.setfield!(obj, val, field)
+@inline traced_setfield!(@nospecialize(obj), field, val) = Base.setfield!(obj, field, val)
 @inline traced_setfield!(@nospecialize(obj::AbstractArray), field, val) =
     Base.setindex!(obj, val, field)
+@inline traced_setfield!(@nospecialize(obj::RArray), field, val) =
+    Base.setfield!(obj, field, val)
 
 function create_result(tocopy::T, path, result_stores) where {T}
     if !isstructtype(typeof(tocopy))

src/Tracing.jl

@@ -12,7 +12,9 @@ for T in (DataType, Module, Nothing, Symbol, AbstractChar, AbstractString, RArra
     end
 end
 
-function traced_type(::Type{T}, seen, mode::Val{Mode}, track_numbers) where {T<:Number,Mode}
+function traced_type(
+    ::Type{T}, seen, mode::Val{Mode}, track_numbers
+) where {T<:Union{AbstractFloat,Integer},Mode}
     if Mode == ArrayToConcrete && any(Base.Fix1(<:, T), track_numbers)
         return ConcreteRNumber{T}
     end

test/tracing.jl

@@ -90,7 +90,7 @@ using Test
                 (Val{:x}, Val{:x}),
             ]
                 tracedty = traced_type(
-                    origty, Reactant.OrderedIdDict(), Val(ConcreteToTraced)
+                    origty, Reactant.OrderedIdDict(), Val(ConcreteToTraced), ()
                 )
                 @test tracedty == targetty
             end
@@ -102,21 +102,21 @@ using Test
                 TracedRArray{Float64,3},
             ]
                 @test_throws Union{ErrorException,String} traced_type(
-                    type, Reactant.OrderedIdDict(), Val(ConcreteToTraced)
+                    type, Reactant.OrderedIdDict(), Val(ConcreteToTraced), ()
                 )
             end
         end
         @testset "traced_type exceptions" begin
             @test_throws TracedTypeError Reactant.traced_type(
-                Real, Reactant.OrderedIdDict(), Val(Reactant.ArrayToConcrete)
+                Real, Reactant.OrderedIdDict(), Val(Reactant.ArrayToConcrete), ()
             )
 
             struct Node
                 x::Vector{Float64}
                 y::Union{Nothing,Node}
             end
             @test_throws NoFieldMatchError traced_type(
-                Node, Reactant.OrderedIdDict(), Val(ArrayToConcrete)
+                Node, Reactant.OrderedIdDict(), Val(ArrayToConcrete), ()
             )
         end
     end