Show for RangeCumsum (#218)

JuliaLinearAlgebra · Mar 22, 2024 · bca3a6b · bca3a6b · jishnub · Mar 22, 2024
1 parent 52d8d29
commit bca3a6b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/cumsum.jl b/src/cumsum.jl
@@ -71,3 +71,5 @@ end
 function Broadcast.broadcasted(::Broadcast.DefaultArrayStyle{1}, ::typeof(*), r::RangeCumsum, x::Number)
     RangeCumsum(r.range * x)
 end
+
+Base.show(io::IO, r::RangeCumsum) = print(io, RangeCumsum, "(", r.range, ")")
diff --git a/test/test_cumsum.jl b/test/test_cumsum.jl
@@ -33,6 +33,7 @@ cmpop(p) = isinteger(real(first(p))) && isinteger(real(step(p))) ? (==) : (≈)
         @test diff(r) == p[firstindex(p)+1:end]
         @test last(r) == r[end] == sum(p)
         @test first(r) == r[firstindex(r)] == first(p)
+        @test repr(r) == "$RangeCumsum($p)"
     end
 
     a,b = RangeCumsum(Base.OneTo(5)), RangeCumsum(Base.OneTo(6))