Skip to content

Commit

Permalink
Implement Iteration interface and Indexing interface for 'SequentialT…
Browse files Browse the repository at this point in the history
…ransform' (#6)
  • Loading branch information
eliascarv committed Jan 24, 2024
1 parent 3ae4084 commit 4a85ce8
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 0 deletions.
9 changes: 9 additions & 0 deletions src/sequential.jl
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,15 @@ Create a [`SequentialTransform`](@ref) transform with
AbstractTrees.nodevalue(::SequentialTransform) = SequentialTransform
AbstractTrees.children(s::SequentialTransform) = s.transforms

# iteration interface
Base.length(s::SequentialTransform) = length(s.transforms)
Base.iterate(s::SequentialTransform, args...) = iterate(s.transforms, args...)

# indexing interface
Base.getindex(s::SequentialTransform, i) = getindex(s.transforms, i)
Base.firstindex(s::SequentialTransform) = firstindex(s.transforms)
Base.lastindex(s::SequentialTransform) = lastindex(s.transforms)

Base.show(io::IO, s::SequentialTransform) = print(io, join(s.transforms, ""))

function Base.show(io::IO, ::MIME"text/plain", s::SequentialTransform)
Expand Down
17 changes: 17 additions & 0 deletions test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -26,4 +26,21 @@ using Test
T = TestTransform() TestTransform()
@test (T Identity()) == T
@test (Identity() T) == T

# sequential
T = TransformsBase.SequentialTransform([TestTransform(), Identity()])
# iteration interface
@test length(T) == 2
T1, state = iterate(T)
@test T1 == TestTransform()
T2, state = iterate(T, state)
@test T2 == Identity()
@test isnothing(iterate(T, state))
# indexing interface
@test T[1] == TestTransform()
@test T[2] == Identity()
@test firstindex(T) == 1
@test lastindex(T) == 2
@test T[begin] == TestTransform()
@test T[end] == Identity()
end

0 comments on commit 4a85ce8

Please sign in to comment.