complex: define unary plus and add tests (JuliaLang#31237)

Liozou · Mar 6, 2019 · 6552521 · 6552521
1 parent 00d6750
commit 6552521
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/base/complex.jl b/base/complex.jl
@@ -262,6 +262,7 @@ abs2(z::Complex) = real(z)*real(z) + imag(z)*imag(z)
 inv(z::Complex) = conj(z)/abs2(z)
 inv(z::Complex{<:Integer}) = inv(float(z))
 
++(z::Complex) = Complex(+real(z), +imag(z))
 -(z::Complex) = Complex(-real(z), -imag(z))
 +(z::Complex, w::Complex) = Complex(real(z) + real(w), imag(z) + imag(w))
 -(z::Complex, w::Complex) = Complex(real(z) - real(w), imag(z) - imag(w))

diff --git a/test/complex.jl b/test/complex.jl
@@ -15,6 +15,17 @@ end
 @test sprint(show, complex(1, 0), context=:compact => true) == "1+0im"
 @test sprint(show, complex(true, true)) == "Complex(true,true)"
 
+@testset "unary operator on complex boolean" begin
+ @test +Complex(true, true) === Complex(1, 1)
+ @test +Complex(true, false) === Complex(1, 0)
+ @test +Complex(false, true) === Complex(0, 1)
+ @test +Complex(false, false) === Complex(0, 0)
+ @test -Complex(true, true) === Complex(-1, -1)
+ @test -Complex(true, false) === Complex(-1, 0)
+ @test -Complex(false, true) === Complex(0, -1)
+ @test -Complex(false, false) === Complex(0, 0)
+end
+
 @testset "arithmetic" begin
  @testset for T in (Float16, Float32, Float64, BigFloat)
  t = true