Skip to content

Commit

Permalink
Fix IteratorSize,IteratorEltype tests
Browse files Browse the repository at this point in the history
  • Loading branch information
mofeing committed Jun 11, 2024
1 parent 4fc90ca commit e61db4f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/Tensor_test.jl
Original file line number Diff line number Diff line change
Expand Up @@ -189,8 +189,8 @@
data = rand(2, 2, 2)
tensor = Tensor(data, (:i, :j, :k))

@test Base.IteratorSize(tensor) == Base.HasShape()
@test Base.IteratorEltype(tensor) == eltype(data)
@test Base.IteratorSize(tensor) == Base.HasShape{3}()
@test Base.IteratorEltype(tensor) == Base.HasEltype()
@test all(x -> ==(x...), zip(tensor, data))
end

Expand Down

0 comments on commit e61db4f

Please sign in to comment.