Allow unzip to return JLArray (#759)

* Allow unzip to return JLArray * style * style Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> * extend comment Co-authored-by: Michael Abbott <[email protected]> --------- Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: Michael Abbott <[email protected]>
JuliaDiff · Dec 1, 2023 · 40b9058 · 40b9058
1 parent 87f4996
commit 40b9058
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/test/unzipped.jl b/test/unzipped.jl
@@ -87,11 +87,14 @@ using ChainRules: unzip_broadcast, unzip #, unzip_map
         # TODO invent some tests of this rrule's pullback function
 
         @test unzip(jl([(1,2), (3,4), (5,6)])) == (jl([1, 3, 5]), jl([2, 4, 6]))
-
         @test unzip(jl([(missing,2), (missing,4), (missing,6)]))[2] == jl([2, 4, 6])
-        @test unzip(jl([(missing,2), (missing,4), (missing,6)]))[2] isa Base.ReinterpretArray
-
         @test unzip(jl([(1,), (3,), (5,)]))[1] == jl([1, 3, 5])
-        @test unzip(jl([(1,), (3,), (5,)]))[1] isa Base.ReinterpretArray
+
+        # depending on Julia/package versions, may get ReinterpretArray or JLArray
+        # Either is acceptable
+        @test isa(
+            unzip(jl([(missing, 2), (missing, 4), (missing, 6)]))[2],
+            Union{Base.ReinterpretArray,JLArray},
+        )
     end
-end
+end