From bac6451443f5cf9995453fc780cd2a8d037e60cd Mon Sep 17 00:00:00 2001
From: Iaroslav <postovalovya@gmail.com>
Date: Sat, 27 Jun 2020 21:17:40 +0700
Subject: [PATCH] Add tests

---
 .../kmath/nd4j/INDArrayStructureTest.kt       | 38 +++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 kmath-nd4j/src/test/kotlin/scientifik/kmath/nd4j/INDArrayStructureTest.kt

diff --git a/kmath-nd4j/src/test/kotlin/scientifik/kmath/nd4j/INDArrayStructureTest.kt b/kmath-nd4j/src/test/kotlin/scientifik/kmath/nd4j/INDArrayStructureTest.kt
new file mode 100644
index 000000000..e851f2e80
--- /dev/null
+++ b/kmath-nd4j/src/test/kotlin/scientifik/kmath/nd4j/INDArrayStructureTest.kt
@@ -0,0 +1,38 @@
+package scientifik.kmath.nd4j
+
+import org.nd4j.linalg.factory.Nd4j
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+internal class INDArrayStructureTest {
+    @Test
+    fun testElements() {
+        val nd = Nd4j.create(doubleArrayOf(1.0, 2.0, 3.0))!!
+        val struct = INDArrayDoubleStructure(nd)
+        val res = struct.elements().map { it.second }.toList()
+        assertEquals(listOf(1.0, 2.0, 3.0), res)
+    }
+
+    @Test
+    fun testShape() {
+        val nd = Nd4j.rand(10, 2, 3, 6)!!
+        val struct = INDArrayIntStructure(nd)
+        assertEquals(intArrayOf(10, 2, 3, 6).toList(), struct.shape.toList())
+    }
+
+    @Test
+    fun testEquals() {
+        val nd1 = Nd4j.create(doubleArrayOf(1.0, 2.0, 3.0))!!
+        val struct1 = INDArrayDoubleStructure(nd1)
+        val nd2 = Nd4j.create(doubleArrayOf(1.0, 2.0, 3.0))!!
+        val struct2 = INDArrayDoubleStructure(nd2)
+        assertEquals(struct1, struct2)
+    }
+
+    @Test
+    fun testDimension() {
+        val nd = Nd4j.rand(8, 16, 3, 7, 1)!!
+        val struct = INDArrayIntStructure(nd)
+        assertEquals(5, struct.dimension)
+    }
+}