import math import cmath // Tests are based on and verified from practice examples of Khan Academy // https://www.khanacademy.org/math/precalculus/imaginary-and-complex-numbers fn test_complex_addition() { mut c1 := cmath.complex(0,-10) mut c2 := cmath.complex(-40,8) mut result := c1 + c2 assert result.equals(cmath.complex(-40,-2)) c1 = cmath.complex(-71,2) c2 = cmath.complex(88,-12) result = c1 + c2 assert result.equals(cmath.complex(17,-10)) c1 = cmath.complex(0,-30) c2 = cmath.complex(52,-30) result = c1 + c2 assert result.equals(cmath.complex(52,-60)) c1 = cmath.complex(12,-9) c2 = cmath.complex(32,-6) result = c1 + c2 assert result.equals(cmath.complex(44,-15)) } fn test_complex_subtraction() { mut c1 := cmath.complex(-8,0) mut c2 := cmath.complex(6,30) mut result := c1 - c2 assert result.equals(cmath.complex(-14,-30)) c1 = cmath.complex(-19,7) c2 = cmath.complex(29,32) result = c1 - c2 assert result.equals(cmath.complex(-48,-25)) c1 = cmath.complex(12,0) c2 = cmath.complex(23,13) result = c1 - c2 assert result.equals(cmath.complex(-11,-13)) c1 = cmath.complex(-14,3) c2 = cmath.complex(0,14) result = c1 - c2 assert result.equals(cmath.complex(-14,-11)) } fn test_complex_multiplication() { mut c1 := cmath.complex(1,2) mut c2 := cmath.complex(1,-4) mut result := c1.multiply(c2) assert result.equals(cmath.complex(9,-2)) c1 = cmath.complex(-4,-4) c2 = cmath.complex(-5,-3) result = c1.multiply(c2) assert result.equals(cmath.complex(8,32)) c1 = cmath.complex(4,4) c2 = cmath.complex(-2,-5) result = c1.multiply(c2) assert result.equals(cmath.complex(12,-28)) c1 = cmath.complex(2,-2) c2 = cmath.complex(4,-4) result = c1.multiply(c2) assert result.equals(cmath.complex(0,-16)) } fn test_complex_division() { mut c1 := cmath.complex(-9,-6) mut c2 := cmath.complex(-3,-2) mut result := c1.divide(c2) assert result.equals(cmath.complex(3,0)) c1 = cmath.complex(-23,11) c2 = cmath.complex(5,1) result = c1.divide(c2) assert result.equals(cmath.complex(-4,3)) c1 = cmath.complex(8,-2) c2 = cmath.complex(-4,1) result = c1.divide(c2) assert result.equals(cmath.complex(-2,0)) c1 = cmath.complex(11,24) c2 = cmath.complex(-4,-1) result = c1.divide(c2) assert result.equals(cmath.complex(-4,-5)) } fn test_complex_conjugate() { mut c1 := cmath.complex(0,8) mut result := c1.conjugate() assert result.equals(cmath.complex(0,-8)) c1 = cmath.complex(7,3) result = c1.conjugate() assert result.equals(cmath.complex(7,-3)) c1 = cmath.complex(2,2) result = c1.conjugate() assert result.equals(cmath.complex(2,-2)) c1 = cmath.complex(7,0) result = c1.conjugate() assert result.equals(cmath.complex(7,0)) } fn test_complex_equals() { mut c1 := cmath.complex(0,8) mut c2 := cmath.complex(0,8) assert c1.equals(c2) c1 = cmath.complex(-3,19) c2 = cmath.complex(-3,19) assert c1.equals(c2) } fn test_complex_abs() { mut c1 := cmath.complex(3,4) assert c1.abs() == 5 c1 = cmath.complex(1,2) assert c1.abs() == math.sqrt(5) assert c1.abs() == c1.conjugate().abs() c1 = cmath.complex(7,0) assert c1.abs() == 7 } fn test_complex_angle(){ mut c := cmath.complex(1, 0) assert c.angle() * 180 / math.Pi == 0 c = cmath.complex(1, 1) assert c.angle() * 180 / math.Pi == 45 c = cmath.complex(0, 1) assert c.angle() * 180 / math.Pi == 90 c = cmath.complex(-1, 1) assert c.angle() * 180 / math.Pi == 135 c = cmath.complex(-1, -1) assert c.angle() * 180 / math.Pi == -135 mut cc := c.conjugate() assert cc.angle() + c.angle() == 0 } fn test_complex_addinv() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(-5,-7) mut result := c1.addinv() assert result.equals(c2) c1 = cmath.complex(-3,4) c2 = cmath.complex(3,-4) result = c1.addinv() assert result.equals(c2) c1 = cmath.complex(-1,-2) c2 = cmath.complex(1,2) result = c1.addinv() assert result.equals(c2) } fn test_complex_mulinv() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(0.067568,-0.094595) mut result := c1.mulinv() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(-0.12,-0.16) result = c1.mulinv() assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-0.2,0.4) result = c1.mulinv() assert result.equals(c2) } fn test_complex_mod() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut result := c1.mod() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq('8.602325') c1 = cmath.complex(-3,4) result = c1.mod() assert result == 5 c1 = cmath.complex(-1,-2) result = c1.mod() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq('2.236068') } fn test_complex_pow() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(-24.0,70.0) mut result := c1.pow(2) // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(117,44) result = c1.pow(3) // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-7,-24) result = c1.pow(4) // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_root() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(2.607904,1.342074) mut result := c1.root(2) // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(1.264953,1.150614) result = c1.root(3) // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(1.068059,-0.595482) result = c1.root(4) // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_exp() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(111.889015,97.505457) mut result := c1.exp() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(-0.032543,-0.037679) result = c1.exp() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-0.153092,-0.334512) result = c1.exp() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_ln() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(2.152033,0.950547) mut result := c1.ln() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(1.609438,2.214297) result = c1.ln() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(0.804719,-2.034444) result = c1.ln() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_sin() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(-525.794515,155.536550) mut result := c1.sin() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(-3.853738,-27.016813) result = c1.sin() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-3.165779,-1.959601) result = c1.sin() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_cos() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(155.536809,525.793641) mut result := c1.cos() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(-27.034946,3.851153) result = c1.cos() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(2.032723,-3.051898) result = c1.cos() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_tan() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(-0.000001,1.000001) mut result := c1.tan() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(0.000187,0.999356) result = c1.tan() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-0.033813,-1.014794) result = c1.tan() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_sinh() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(55.941968,48.754942) mut result := c1.sinh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(6.548120,-7.619232) result = c1.sinh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(0.489056,-1.403119) result = c1.sinh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_cosh() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(55.947047,48.750515) mut result := c1.cosh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(-6.580663,7.581553) result = c1.cosh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-0.642148,1.068607) result = c1.cosh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) } fn test_complex_tanh() { // Tests were also verified on Wolfram Alpha mut c1 := cmath.complex(5,7) mut c2 := cmath.complex(0.999988,0.000090) mut result := c1.tanh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-3,4) c2 = cmath.complex(-1.000710,0.004908) result = c1.tanh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) c1 = cmath.complex(-1,-2) c2 = cmath.complex(-1.166736,0.243458) result = c1.tanh() // Some issue with precision comparison in f64 using == operator hence serializing to string assert result.str().eq(c2.str()) }