fixed AssertionError (#160)

* fixed AssertionError

* fixed

* few changes

* fixed

* added test

* added test

* Update test/transforms/zscore.jl

Co-authored-by: Elias Carvalho <[email protected]>

Co-authored-by: Elias Carvalho <[email protected]>

Author: vickydeka

src/transforms.jl

@@ -170,16 +170,17 @@ reapply(transform::StatelessFeatureTransform, table, cache) =
 # ------------------
 
 function applyfeat(transform::ColwiseFeatureTransform, feat, prep)
-  # basic checks
-  for assertion in assertions(transform)
-    assertion(feat)
-  end
-
   # retrieve column names and values
   cols   = Tables.columns(feat)
   names  = Tables.columnnames(cols)
   snames = choose(transform.colspec, names)
+  sfeat  = feat |> Select(snames)
 
+  # basic checks
+  for assertion in assertions(transform)
+    assertion(sfeat)
+  end
+  
   # function to transform a single column
   function colfunc(n)
     x = Tables.getcolumn(cols, n)
@@ -288,4 +289,4 @@ include("transforms/eigenanalysis.jl")
 include("transforms/projectionpursuit.jl")
 include("transforms/rowtable.jl")
 include("transforms/coltable.jl")
-include("transforms/parallel.jl")
+include("transforms/parallel.jl")

test/transforms/zscore.jl

@@ -86,4 +86,14 @@
   @test isapprox(σ[2], 1; atol=1e-6)
   tₒ = revert(T, n, c)
   @test Tables.matrix(t) ≈ Tables.matrix(tₒ)
+  
+  a = rand(Int, 10)
+  b = rand(10)
+  t = Table(; a, b)
+  T = ZScore(:b)
+  n, c = apply(T, t)
+  @test isapprox(mean(n.b), 0; atol=1e-6)
+  @test isapprox(std(n.b), 1; atol=1e-6)
+  tₒ = revert(T, n, c)
+  @test Tables.matrix(t) ≈ Tables.matrix(tₒ)
 end