Added test for #24 issue

Kotlin · Jan 14, 2021 · bfc3244 · bfc3244
1 parent 964442b
commit bfc3244
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/integration/InferenceModelTest.kt b/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/integration/InferenceModelTest.kt
@@ -220,9 +220,18 @@ class InferenceModelTest {
                     it.predict(train.getX(0))
                 }
             assertEquals(
-                "Reshape functions is missed!",
+                "Reshape functions is missed! Define and set up the reshape function to transform initial data to the model input.",
                 exception.message
             )
+
+            val exception2 =
+                Assertions.assertThrows(IllegalArgumentException::class.java) {
+                    it.predictSoftly(train.getX(0))
+                }
+            assertEquals(
+                "Reshape functions is missed! Define and set up the reshape function to transform initial data to the model input.",
+                exception2.message
+            )
         }
     }