add test cases

JuliaFolds2 · Jan 29, 2024 · 590e058 · 590e058
1 parent 02c39c2
commit 590e058
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -2,7 +2,7 @@ using Test, ThreadsBasics
 
 @testset "Basics" begin
     for (~, f, op, itr) ∈ [
-        (isapprox, sin, +, rand(ComplexF64, 100)),
+        (isapprox, sin, +, rand(ComplexF64, 10, 10)),
         (isapprox, cos, max, 1:100000),
         (==, round, vcat, randn(1000)),
         (==, last, *, [1=>"a", 2=>"b", 3=>"c", 4=>"d", 5=>"e"])
@@ -25,6 +25,10 @@ using Test, ThreadsBasics
                     @test all(tmap(f, Any, itr; kwargs...) .~ map_f_itr)
                     @test all(tcollect(Any, (f(x) for x in itr); kwargs...) .~ map_f_itr)
                     @test all(tcollect(Any, f.(itr); kwargs...) .~ map_f_itr)
+
+                    @test tmap(f, itr; kwargs...) ~ map_f_itr
+                    @test tcollect((f(x) for x in itr); kwargs...) ~ map_f_itr
+                    @test tcollect(f.(itr); kwargs...) ~ map_f_itr
 
                     RT = Core.Compiler.return_type(f, Tuple{eltype(itr)})