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

Added extension to field to use numbers as first class citizens

Browse Source
This commit is contained in:
Alexander Nozik 2018-08-25 21:15:55 +03:00
parent c344380120
commit 3576ddaf31
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kmath-common/src/main/kotlin/scientifik/kmath/operations/Algebra.kt
View File

@ -101,6 +101,9 @@ interface Field<T> : Ring<T> {
operator fun T.div(b: T): T = divide(this, b) operator fun T.div(b: T): T = divide(this, b)
operator fun Number.div(b: T) = this * divide(one, b) operator fun Number.div(b: T) = this * divide(one, b)
operator fun T.plus(b: Number) = this.plus(b * one)
operator fun Number.plus(b: T) = b + this
} }
/** /**

2
kmath-common/src/test/kotlin/scientifik/kmath/expressions/FieldExpressionContextTest.kt
View File

@ -10,7 +10,7 @@ class FieldExpressionContextTest {
val context = FieldExpressionContext(DoubleField) val context = FieldExpressionContext(DoubleField)
val expression = with(context) { val expression = with(context) {
val x = variable("x", 2.0) val x = variable("x", 2.0)
x * x + 2 * x + 1.0 * one x * x + 2 * x + 1.0
} }
assertEquals(expression("x" to 1.0), 4.0) assertEquals(expression("x" to 1.0), 4.0)
assertEquals(expression(), 9.0) assertEquals(expression(), 9.0)