Merge pull request #67 from JuliaAlgebra/bl/round

Add arguments to round

Author: blegat

src/polynomial.jl

@@ -368,8 +368,8 @@ function mapcoefficientsnz(f::Function, p::AbstractPolynomialLike)
 end
 mapcoefficientsnz(f::Function, t::AbstractTermLike) = f(coefficient(t)) * monomial(t)
 
-Base.round(t::AbstractTermLike) = round(coefficient(t)) * monomial(t)
-function Base.round(p::AbstractPolynomialLike)
+Base.round(t::AbstractTermLike, args...) = round(coefficient(t), args...) * monomial(t)
+function Base.round(p::AbstractPolynomialLike, args...)
     # round(0.1) is zero so we cannot use SortedUniqState
-    polynomial(round.(terms(p)), SortedState())
+    polynomial(round.(terms(p), args...), SortedState())
 end

test/polynomial.jl

@@ -117,6 +117,8 @@
 
     @test (@inferred round(2.6x + 1.001x^2)) == 3x + 1x^2
     @test (@inferred round(3.1x*y)) == 3x*y
+    @test (@inferred round(2.613x + 1.1051x^2, 2)) ≈ 2.61x + 1.11x^2
+    @test (@inferred round(3.145x*y, 1)) ≈ 3.1x*y
 
     @testset "Graded Lex Order" begin
         Mod.@polyvar x y z