Add KeyedArray and AxisArray tests

test/runtests.jl

@@ -27,4 +27,5 @@ using TimeZones
     include("types/tables.jl")
     include("types/vector.jl")
     include("types/matrix.jl")
+    include("types/xarray.jl")
 end

test/types/xarray.jl

@@ -0,0 +1,124 @@
+@testset "$ArrayType" for ArrayType in (AxisArray, KeyedArray)
+
+    # TODO: test on dims=:foo, :bar
+
+    @testset "is_transformable" begin
+        x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+        @test is_transformable(x)
+    end
+
+    @testset "apply" begin
+
+        @testset "OneToOne" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeOneToOneTransform()
+            @test FeatureTransforms.apply(x, T) == ones(2, 3)
+            @test FeatureTransforms.apply(x, T; inds=[1, 2]) == ones(2)
+            @test FeatureTransforms.apply(x, T; dims=2, inds=[2, 3]) == ones(2, 2)
+        end
+
+        @testset "OneToMany" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeOneToManyTransform()
+            @test FeatureTransforms.apply(x, T) == ones(2, 6)
+            @test FeatureTransforms.apply(x, T; inds=[1, 2]) == ones(2, 2)
+            @test FeatureTransforms.apply(x, T; dims=2, inds=[2, 3]) == ones(2, 4)
+        end
+
+        @testset "ManyToOne" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeManyToOneTransform()
+            @test FeatureTransforms.apply(x, T; dims=1) == ones(3)
+            @test FeatureTransforms.apply(x, T; dims=1, inds=[1, 2]) == ones(3)
+            @test FeatureTransforms.apply(x, T; dims=2, inds=[1, 2]) == ones(2)
+        end
+
+        @testset "ManyToMany" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeManyToManyTransform()
+            @test FeatureTransforms.apply(x, T; dims=1) == ones(2, 6)
+            @test FeatureTransforms.apply(x, T; inds=[1, 2]) == ones(2, 2)
+            @test FeatureTransforms.apply(x, T; dims=2, inds=[2, 3]) == ones(2, 4)
+        end
+
+    end
+
+    @testset "apply!" begin
+        T = FakeOneToOneTransform()
+
+        x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+        FeatureTransforms.apply!(x, T)
+        @test x == ones(2, 3)
+
+        # https://github.com/invenia/FeatureTransforms.jl/issues/68
+        x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+        @test_broken FeatureTransforms.apply!(x, T; inds=[1, 2])
+    end
+
+    @testset "apply_append" begin
+
+        @testset "OneToOne" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeOneToOneTransform()
+
+            @test FeatureTransforms.apply_append(x, T; append_dim=1) == vcat(x, ones(2, 3))
+            @test FeatureTransforms.apply_append(x, T; append_dim=2) == hcat(x, ones(2, 3))
+            @test FeatureTransforms.apply_append(x, T; append_dim=3) == cat(x, ones(2, 3), dims=3)
+
+            @test isequal(
+                FeatureTransforms.apply_append(x, T; dims=1, inds=[1], append_dim=1),
+                vcat(x, ones(1, 3))
+            )
+
+            @test isequal(
+                FeatureTransforms.apply_append(x, T; dims=2, inds=[1, 2], append_dim=2),
+                hcat(x, ones(2, 2))
+            )
+        end
+
+        @testset "OneToMany" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeOneToManyTransform()
+
+            @test_throws DimensionMismatch FeatureTransforms.apply_append(x, T; append_dim=1)
+            @test_throws DimensionMismatch FeatureTransforms.apply_append(x, T; append_dim=3)
+
+            @test isequal(
+                FeatureTransforms.apply_append(x, T; append_dim=2),
+                hcat(x, ones(2, 6))
+            )
+
+            @test isequal(
+                FeatureTransforms.apply_append(x, T; inds=[1, 2], append_dim=2),
+                hcat(x, ones(2, 2))
+            )
+        end
+
+        @testset "ManyToOne" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeManyToOneTransform()
+            @test FeatureTransforms.apply_append(x, T; dims=1, append_dim=1) == vcat(x, ones(1, 3))
+            @test FeatureTransforms.apply_append(x, T; dims=2, append_dim=2) == hcat(x, ones(2))
+        end
+
+        @testset "ManyToMany" begin
+            x = ArrayType([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            T = FakeManyToManyTransform()
+
+            @test_throws DimensionMismatch FeatureTransforms.apply_append(x, T; append_dim=1)
+            @test_throws DimensionMismatch FeatureTransforms.apply_append(x, T; append_dim=3)
+
+            @test isequal(
+                FeatureTransforms.apply_append(x, T; append_dim=2),
+                hcat(x, ones(2, 6))
+            )
+
+            @test isequal(
+                FeatureTransforms.apply_append(x, T; inds=[1, 2], append_dim=2),
+                hcat(x, ones(2, 2))
+            )
+        end
+
+    end
+
+end