test on Reshaped and Sub arrays

invenia · Jul 1, 2021 · cdb209a · cdb209a
1 parent bd8d4f9
commit cdb209a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,5 +1,6 @@
 using AxisArrays
 using AxisKeys
+using Base: ReshapedArray
 using DataFrames: DataFrame
 using Dates
 using Documenter: doctest

diff --git a/test/types/matrix.jl b/test/types/matrix.jl
@@ -1,6 +1,11 @@
-@testset "matrix" begin
+# Want to test that certain subtypes of arrays are compatible
+_make_matrix(::Type{Matrix}) = [1 2 3; 4 5 6]
+_make_matrix(::Type{SubArray}) = view([1 2 3; 4 5 6], :, :)
+_make_matrix(::Type{ReshapedArray}) = reshape([1 2 3; 4 5 6], 2, 3)
 
-    x = [1 2 3; 4 5 6]
+@testset "$MatrixType" for MatrixType in (Matrix, SubArray, ReshapedArray)
+
+    x = _make_matrix(MatrixType)
 
     @test is_transformable(x)