Merge 24250a8 into 9ba45f8

Jutho · Jan 7, 2021 · dde607b · dde607b
2 parents 9ba45f8 + 24250a8
commit dde607b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/RationalRoots.jl b/src/RationalRoots.jl
@@ -154,6 +154,9 @@ for op in (:*, :/, :\, ://)
         signedroot($op(signedsquare(x), signedsquare(y)))
 end
 
+# When squared, return Rational type:
+Base.literal_pow(::typeof(^), x::RationalRoot, ::Val{2}) = abs(x.signedsquare)
+
 Base.inv(x::RationalRoot) = signedroot(inv(signedsquare(x)))
 
 Base.one(x::RationalRoot) = one(typeof(x))

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -92,6 +92,7 @@ end
         @test iszero(zero(x))
         @test !isone(zero(x))
         @test zero(x) * x === zero(x)
+        @test x^2 === abs(a)
     end
 end