Compact show for ZeroBundle etc (#136)

* Compact show for ZeroBundle etc

* Also overload show for types themselves

* Fix test

---------

Co-authored-by: Keno Fischer <[email protected]>

Author: oxinabox

src/tangent.jl

@@ -250,8 +250,34 @@ UniformBundle{N}(primal, partial::U) where {N,U} = _TangentBundle(Val{N}(), prim
 UniformBundle{N, <:Any, U}(primal, partial::U) where {N, U} = _TangentBundle(Val{N}(), primal, UniformTangent{U}(U.instance))
 UniformBundle{N, <:Any, U}(primal) where {N, U} = _TangentBundle(Val{N}(), primal, UniformTangent{U}(U.instance))
 
+
 const ZeroBundle{N, B} = UniformBundle{N, B, ZeroTangent}
 const DNEBundle{N, B} = UniformBundle{N, B, NoTangent}
+const AbstractZeroBundle{N, B} = UniformBundle{N, B, <:AbstractZero}
+
+wrapper_name(::Type{<:ZeroBundle}) = "ZeroBundle"
+wrapper_name(::Type{<:DNEBundle}) = "DNEBundle"
+wrapper_name(::Type{<:AbstractZeroBundle}) = "AbstractZeroBundle"
+
+function Base.show(io::IO, T::Type{<:AbstractZeroBundle{N, B}}) where {N,B}
+    print(io, wrapper_name(T))
+    print(io, "{$N, ")
+    show(io, B)
+    print(io, "}")
+end
+
+function Base.show(io::IO, T::Type{<:AbstractZeroBundle{N}}) where {N}
+    print(io, wrapper_name(T))
+    print(io, "{$N}")
+end
+
+function Base.show(io::IO, t::AbstractZeroBundle{N}) where N
+    print(io, wrapper_name(typeof(t)))
+    print(io, "{$N}(")
+    show(io, t.primal)
+    print(io, ")")
+end
+
 
 Base.getindex(u::UniformBundle, ::TaylorTangentIndex) = u.tangent.val
 

test/runtests.jl

@@ -13,7 +13,9 @@ const bwd = Diffractor.PrimeDerivativeBack
 
 @testset verbose=true "Diffractor.jl" begin  # overall testset, ensures all tests run
 
-include("stage2_fwd.jl")
+@testset "$file" for file in ("stage2_fwd.jl", "tangent.jl")
+    include(file)
+end
 
 # Unit tests
 function tup2(f)

test/tangent.jl

@@ -0,0 +1,25 @@
+module tagent
+using Diffractor
+using Diffractor: AbstractZeroBundle, ZeroBundle, DNEBundle
+using Test
+
+@testset "AbstractZeroBundle" begin
+    @testset "Hierachy" begin
+        @test ZeroBundle <: AbstractZeroBundle
+        @test DNEBundle <: AbstractZeroBundle
+        @test ZeroBundle{1} <: AbstractZeroBundle{1}
+        @test ZeroBundle{1,typeof(getfield)} <: AbstractZeroBundle{1,typeof(getfield)}
+    end
+
+    @testset "Display" begin
+        @test repr(ZeroBundle{1}(2.0)) == "ZeroBundle{1}(2.0)"
+        @test repr(DNEBundle{1}(getfield)) == "DNEBundle{1}(getfield)"
+
+        @test repr(ZeroBundle{1}) == "ZeroBundle{1}"
+        @test repr(ZeroBundle{1, Float64}) == "ZeroBundle{1, Float64}"
+
+        @test repr(typeof(DNEBundle{1}(getfield))) == "DNEBundle{1, typeof(getfield)}"
+    end
+end
+
+end  # module