Test application of IdentityScaling() does nothing (#113)

* Test application of `IdentityScaling()` does nothing * Test `DataFrame` and `NamedDimsArray`
invenia · Oct 31, 2022 · ee13296 · ee13296
1 parent cc299dc
commit ee13296
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/scaling.jl b/test/scaling.jl
@@ -9,6 +9,20 @@
             @test IdentityScaling(123) == IdentityScaling()
             @test IdentityScaling([1, 2, 3]) == IdentityScaling()
         end
+
+        @testset "Transform does nothing" begin
+            x = ones(2, 3, 4)
+            @test FeatureTransforms.apply(x, IdentityScaling()) == x
+            @test FeatureTransforms.apply(x, IdentityScaling()) isa Array
+
+            table = DataFrame((a=[1, 2, 3], b=[4, 5, 6]))
+            @test FeatureTransforms.apply(table, IdentityScaling(); header=(:a, :b)) == table
+            @test FeatureTransforms.apply(table, IdentityScaling(); header=(:a, :b)) isa DataFrame
+
+            na = NamedDimsArray([1 2 3; 4 5 6], foo=["a", "b"], bar=[:x, :y, :z])
+            @test FeatureTransforms.apply(na, IdentityScaling()) == na
+            @test FeatureTransforms.apply(na, IdentityScaling()) isa NamedDimsArray
+        end
     end
 
     @testset "StandardScaling" begin