kmath/kmath-dimensions/src/commonTest/kotlin/kscience/dimensions/DMatrixContextTest.kt

37 lines
1.0 KiB
Kotlin
Raw Normal View History

/*
* Copyright 2018-2021 KMath contributors.
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
*/
2021-03-30 22:24:21 +03:00
package space.kscience.dimensions
2019-06-21 12:34:04 +03:00
2021-02-18 11:17:28 +03:00
import space.kscience.kmath.dimensions.D2
import space.kscience.kmath.dimensions.D3
import space.kscience.kmath.dimensions.DMatrixContext
import space.kscience.kmath.dimensions.one
2019-06-21 12:34:04 +03:00
import kotlin.test.Test
2021-01-19 22:24:42 +03:00
@Suppress("UNUSED_VARIABLE")
2020-09-09 17:12:18 +03:00
internal class DMatrixContextTest {
2019-06-21 12:34:04 +03:00
@Test
fun testDimensionSafeMatrix() {
val res = with(DMatrixContext.real) {
2019-06-21 12:34:04 +03:00
val m = produce<D2, D2> { i, j -> (i + j).toDouble() }
//The dimension of `one()` is inferred from type
(m + one())
}
}
@Test
fun testTypeCheck() {
val res = with(DMatrixContext.real) {
2019-06-21 12:34:04 +03:00
val m1 = produce<D2, D3> { i, j -> (i + j).toDouble() }
val m2 = produce<D3, D2> { i, j -> (i + j).toDouble() }
//Dimension-safe addition
m1.transpose() + m2
}
}
2020-09-09 17:12:18 +03:00
}