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 {
|
private val strides: List<Int> by lazy {
|
||||||
ArrayList<Int>(shape.size).apply {
|
ArrayList<Int>(shape.size).apply {
|
||||||
var current = 1
|
var current = 1
|
||||||
add(0)
|
add(1)
|
||||||
shape.forEach {
|
shape.forEach {
|
||||||
current *= it
|
current *= it
|
||||||
add(current)
|
add(current)
|
||||||
|
@ -18,4 +18,17 @@ class RealNDFieldTest {
|
|||||||
val product = array1*array2
|
val product = array1*array2
|
||||||
assertEquals(0.0, product[2, 2].toDouble(), 0.1)
|
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