jvm/src/main/kotlin/scientifik/kmath/structures/RealNDArray.kt

@@ -12,7 +12,7 @@ private class RealNDField(shape: List<Int>) : NDField<Real>(shape, RealField) {
     private val strides: List<Int> by lazy {
         ArrayList<Int>(shape.size).apply {
             var current = 1
-            add(0)
+            add(1)
             shape.forEach {
                 current *= it
                 add(current)

jvm/src/test/kotlin/scientifik/kmath/structures/RealNDFieldTest.kt

@@ -18,4 +18,17 @@ class RealNDFieldTest {
         val product = array1*array2
         assertEquals(0.0, product[2, 2].toDouble(), 0.1)
     }
+
+    @Test
+    fun testGeneration() {
+
+        val array = real2DArray(3, 3) { i, j -> (i * 10 + j).toDouble() }
+
+        for(i in 0..2){
+            for(j in 0..2){
+                val expected= (i * 10 + j).toDouble()
+                assertEquals("Error at index [$i, $j]", expected, array[i,j].value)
+            }
+        }
+    }
 }