From 04914605542f180d44f7254ae64529c51598436e Mon Sep 17 00:00:00 2001
From: Julia Beliaeva <Julia.Beliaeva@jetbrains.com>
Date: Fri, 22 Apr 2022 01:28:40 +0300
Subject: [PATCH] Test Dense layer export

---
 ...tTest.kt => CoreLayersImportExportTest.kt} | 28 ++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)
 rename api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/{InputLayerImportExportTest.kt => CoreLayersImportExportTest.kt} (52%)

diff --git a/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/InputLayerImportExportTest.kt b/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/CoreLayersImportExportTest.kt
similarity index 52%
rename from api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/InputLayerImportExportTest.kt
rename to api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/CoreLayersImportExportTest.kt
index 626a03154..afddd0e59 100644
--- a/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/InputLayerImportExportTest.kt
+++ b/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/CoreLayersImportExportTest.kt
@@ -7,10 +7,16 @@ package org.jetbrains.kotlinx.dl.api.inference.keras
 
 import org.jetbrains.kotlinx.dl.api.core.Functional
 import org.jetbrains.kotlinx.dl.api.core.Sequential
+import org.jetbrains.kotlinx.dl.api.core.activation.Activations
+import org.jetbrains.kotlinx.dl.api.core.initializer.HeNormal
+import org.jetbrains.kotlinx.dl.api.core.initializer.HeUniform
+import org.jetbrains.kotlinx.dl.api.core.layer.core.Dense
 import org.jetbrains.kotlinx.dl.api.core.layer.core.Input
+import org.jetbrains.kotlinx.dl.api.core.regularizer.L2
+import org.jetbrains.kotlinx.dl.api.core.regularizer.L2L1
 import org.junit.jupiter.api.Test
 
-class InputLayerImportExportTest {
+class CoreLayersImportExportTest {
     @Test
     fun inputLayerSequential() {
         LayerImportExportTest.run(Sequential.of(Input(4)))
@@ -26,4 +32,24 @@ class InputLayerImportExportTest {
         LayerImportExportTest.run(Functional.of(Input(128, 128, 3)))
         LayerImportExportTest.run(Functional.of(Input(10, 10, 10, 10)))
     }
+
+    @Test
+    fun denseLayer() {
+        LayerImportExportTest.run(
+            Sequential.of(
+                Input(10),
+                Dense(
+                    name = "test_dense",
+                    outputSize = 10,
+                    activation = Activations.Tanh,
+                    kernelInitializer = HeNormal(),
+                    biasInitializer = HeUniform(),
+                    kernelRegularizer = L2(),
+                    biasRegularizer = L2(),
+                    activityRegularizer = L2L1(),
+                    useBias = true
+                )
+            )
+        )
+    }
 }