forked from kscience/kmath
Add more corner cases for complex power
This commit is contained in:
parent
83d9e1f0af
commit
ca9df8a167
@ -139,12 +139,19 @@ public object ComplexField :
|
||||
zero
|
||||
}
|
||||
}
|
||||
|
||||
} else {
|
||||
exp(pow * ln(arg))
|
||||
}
|
||||
|
||||
public fun power(arg: Complex, pow: Complex): Complex = exp(pow * ln(arg))
|
||||
public fun power(arg: Complex, pow: Complex): Complex = if(arg == zero || arg == (-0.0).toComplex()){
|
||||
if(pow == zero){
|
||||
one
|
||||
} else {
|
||||
zero
|
||||
}
|
||||
} else {
|
||||
exp(pow * ln(arg))
|
||||
}
|
||||
|
||||
public fun Complex.pow(power: Complex): Complex = power(this, power)
|
||||
|
||||
|
@ -71,6 +71,11 @@ internal class ComplexFieldTest {
|
||||
(i * 8).let { it.im.toInt() to it.re.toInt() },
|
||||
(Complex(2, 2) pow 2).let { it.im.toInt() to it.re.toInt() })
|
||||
|
||||
assertEquals(1.0, Complex(0.0).pow(Complex(0.0)).re, 0.01)
|
||||
assertEquals(1.0, Complex(-0.0).pow(Complex(0.0)).re, 0.01)
|
||||
assertEquals(0.0, Complex(0.0).pow(Complex(2.0)).re, 0.01)
|
||||
assertEquals(0.0, Complex(-0.0).pow(Complex(2.0)).re, 0.01)
|
||||
assertEquals(1.0, Complex(-1.0).pow(Complex(2.0)).re, 0.01)
|
||||
assertEquals(2.0, power(Complex(-4.0, 0.0), 0.5 + 0 * i).im, 0.01)
|
||||
assertEquals(2.0, power(Complex(-4.0, 0.0), 0.5).im, 0.01)
|
||||
}
|
||||
|
@ -19,19 +19,13 @@ import org.ejml.sparse.csc.factory.DecompositionFactory_DSCC
|
||||
import org.ejml.sparse.csc.factory.DecompositionFactory_FSCC
|
||||
import org.ejml.sparse.csc.factory.LinearSolverFactory_DSCC
|
||||
import org.ejml.sparse.csc.factory.LinearSolverFactory_FSCC
|
||||
import space.kscience.kmath.UnstableKMathAPI
|
||||
import space.kscience.kmath.linear.*
|
||||
import space.kscience.kmath.linear.Matrix
|
||||
import space.kscience.kmath.UnstableKMathAPI
|
||||
import space.kscience.kmath.nd.StructureFeature
|
||||
import space.kscience.kmath.structures.Float64
|
||||
import space.kscience.kmath.structures.Float32
|
||||
import space.kscience.kmath.operations.Float64Field
|
||||
import space.kscience.kmath.operations.Float32Field
|
||||
import space.kscience.kmath.operations.DoubleField
|
||||
import space.kscience.kmath.operations.FloatField
|
||||
import space.kscience.kmath.operations.Float64Field
|
||||
import space.kscience.kmath.operations.invoke
|
||||
import space.kscience.kmath.structures.Float64Buffer
|
||||
import space.kscience.kmath.structures.Float32Buffer
|
||||
import space.kscience.kmath.structures.DoubleBuffer
|
||||
import space.kscience.kmath.structures.FloatBuffer
|
||||
import kotlin.reflect.KClass
|
||||
|
Loading…
Reference in New Issue
Block a user