kscience/kmath
9
0
Fork 1
Code Issues Pull Requests 6 Packages Projects Releases 16 Wiki Activity

added test for StructureND<Double>.times

Browse Source
This commit is contained in:
Margarita 2022-03-23 00:46:52 +03:00 committed by margarita0303
parent cc00e1dee5
commit 487a663d0f
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
kmath-tensors/src/commonTest/kotlin/space/kscience/kmath/tensors/core/TestDoubleTensor.kt
View File

@ -49,6 +49,18 @@ internal class TestDoubleTensor {
assertTrue { tensor.rowsByIndices(intArrayOf(0, 1)) eq tensor }
}
@Test
fun testTimes() = DoubleTensorAlgebra {
val shape = intArrayOf(2, 2)
val buffer = doubleArrayOf(1.0, 2.0, -3.0, 4.0)
val tensor = DoubleTensor(shape, buffer)
val value = 3
assertTrue { tensor.times(value).toBufferedTensor() eq DoubleTensor(shape, buffer.map { x -> 3 * x }.toDoubleArray()) }
val buffer2 = doubleArrayOf(7.0, -8.0, -5.0, 2.0)
val tensor2 = DoubleTensor(shape, buffer2)
assertTrue {tensor.times(tensor2).toBufferedTensor() eq DoubleTensor(shape, doubleArrayOf(7.0, -16.0, 15.0, 8.0)) }
}
@Test
fun testValue() = DoubleTensorAlgebra {
val value = 12.5