test summary

KnetML · May 15, 2023 · e333271 · e333271 · andreasdominik · May 15, 2023
1 parent 4c87713
commit e333271
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/test/layers.jl b/test/layers.jl
@@ -346,6 +346,24 @@ function test_summary()
     return n == 15
 end
 
+# test summary for abstract layer:
+#
+mutable struct AL <: AbstractLayer
+    w
+    function AL(i,o)
+        w = param(o,i)
+        return new(w)
+    end
+end
+
+function test_abstract_layer_summary()
+
+    al = AL(100,100)
+    n = summary(al)
+    return n == 1
+end
+
+
 function test_print()
 
     ch = Chain(Conv(3, 3, 3, 100),

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -107,6 +107,7 @@ using Statistics: mean
 @test test_get_set_rnn()
 
 @test test_summary()
+@test test_abstract_layer_summary()
 @test test_print()