Merge pull request #4 from Wasabi375/master
Fixed Stride for RealNDArray
This commit is contained in:
commit
6e2ad178f1
@ -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)
|
||||
|
@ -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)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user