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

Add more test cases for tensor operations #475

Merged
margarita0303 merged 4 commits from feature/tensors-performance into feature/tensors-performance 2022-04-06 14:32:56 +03:00
Conversation 0 Commits 4 Files Changed 2 +26
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 93b62c5bf6 - Show all commits

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

@ -23,6 +23,32 @@ import kotlin.test.assertTrue
internal class TestDoubleTensor { internal class TestDoubleTensor {
@Test
fun testFullLike() = DoubleTensorAlgebra {
val shape = intArrayOf(2, 3)
val buffer = doubleArrayOf(1.0, 2.0, 3.0, 4.0, 5.0, 6.0)
val tensor = DoubleTensor(shape, buffer)
val value = 12.5
assertTrue { tensor.fullLike(value) eq DoubleTensor(shape, buffer.map { value }.toDoubleArray() ) }
}
@Test
fun testOnesLike() = DoubleTensorAlgebra {
val shape = intArrayOf(2, 3)
val buffer = doubleArrayOf(1.0, 2.0, 3.0, 4.0, 5.0, 6.0)
val tensor = DoubleTensor(shape, buffer)
assertTrue { tensor.onesLike() eq DoubleTensor(shape, buffer.map { 1.0 }.toDoubleArray() ) }
}
@Test
fun testRowsByIndices() = DoubleTensorAlgebra {
val shape = intArrayOf(2, 2)
val buffer = doubleArrayOf(1.0, 2.0, -3.0, 4.0)
val tensor = fromArray(shape, buffer)
assertTrue { tensor.rowsByIndices(intArrayOf(0)) eq DoubleTensor(intArrayOf(1, 2), doubleArrayOf(1.0, 2.0)) }
assertTrue { tensor.rowsByIndices(intArrayOf(0, 1)) eq tensor }
}
@Test @Test
fun testValue() = DoubleTensorAlgebra { fun testValue() = DoubleTensorAlgebra {
val value = 12.5 val value = 12.5