Merge pull request #106 from JuliaAlgebra/bl/printapl

🐛 Fix printing of AbstractPolynomialLike

Author: blegat

src/show.jl

@@ -1,5 +1,6 @@
-function Base.show(io::IO, mime::MIME"text/latex",
-                   p::Union{AbstractPolynomialLike, RationalPoly})
+const TypesWithShow = Union{AbstractVariable, AbstractMonomial, AbstractTerm,
+                            AbstractPolynomial, RationalPoly}
+function Base.show(io::IO, mime::MIME"text/latex", p::TypesWithShow)
     print(io, "\$\$ ")
     _show(io, mime, p)
     print(io, " \$\$")
@@ -8,17 +9,15 @@ end
 # If the MIME is not specified, IJulia thinks that it supports images, ...
 # and then use the result of show and tries to interpret it as an svg, ...
 # We need the two methods to avoid ambiguity
-function Base.show(io::IO, mime::MIME"text/plain",
-                   p::Union{AbstractPolynomialLike, RationalPoly})
+function Base.show(io::IO, mime::MIME"text/plain", p::TypesWithShow)
     _show(io, mime, p)
 end
-function Base.show(io::IO, mime::MIME"text/print",
-                   p::Union{AbstractPolynomialLike, RationalPoly})
+function Base.show(io::IO, mime::MIME"text/print", p::TypesWithShow)
     _show(io, mime, p)
 end
 
-Base.print(io::IO, p::Union{AbstractPolynomialLike, RationalPoly}) = show(io, MIME"text/print"(), p)
-Base.show(io::IO, p::Union{AbstractPolynomialLike, RationalPoly}) = show(io, MIME"text/plain"(), p)
+Base.print(io::IO, p::TypesWithShow) = show(io, MIME"text/print"(), p)
+Base.show(io::IO, p::TypesWithShow) = show(io, MIME"text/plain"(), p)
 
 # VARIABLES
 function _show(io::IO, mime::MIME, var::AbstractVariable)

test/show.jl

@@ -3,6 +3,7 @@
     @test sprint(show, (x*y^2 + x + 1 + y)) == "xy² + x + y + 1"
     @test sprint(show, (x + 1 + y) / x^2) == "(x + y + 1) / (x²)"
     @test sprint(show, (x - y - x + y) / (x^2 - x)) == "(0) / (x² - x)"
+    @test sprint(show, CustomPoly(1 + x)) == "CustomPoly{$Int,$(typeof(1 + x))}(x + 1)"
     # Test taken from TypedPolynomials
     @test sprint(show, x) == "x"
     @test sprint(show, x^0) == "1"