tests to ensure proper array type handling

JuliaObjects · Dec 27, 2023 · a10576b · a10576b
1 parent 4653ccc
commit a10576b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_functionlenses.jl b/test/test_functionlenses.jl
@@ -3,6 +3,7 @@ using Test
 using Dates
 using Unitful
 using LinearAlgebra: norm
+using AxisKeys
 using InverseFunctions: inverse
 using Accessors: test_getset_laws, test_modify_law
 using Accessors
@@ -170,6 +171,10 @@ end
     B = @set reverse(vec(A)) = 1:6
     @test B == [6 4 2; 5 3 1]
 
+    C = KeyedArray([1,2,3], x=[:a, :b, :c])
+    @test (@set vec(C) = [5,6,7])::KeyedArray == KeyedArray([5,6,7], x=[:a, :b, :c])
+    @test (@set reverse(C) = [5,6,7])::KeyedArray == KeyedArray([7,6,5], x=[:a, :b, :c])
+
     test_getset_laws(size, A, (1, 6), (3, 2))
     test_getset_laws(vec, A, 10:15, 21:26)
     test_getset_laws(reverse, collect(1:6), 10:15, 21:26)