From 06cf88ba823c74fa93f4d196609493529bdf87ae Mon Sep 17 00:00:00 2001 From: N5N3 <2642243996@qq.com> Date: Mon, 10 Jan 2022 17:46:33 +0800 Subject: [PATCH] add test --- test/broadcast.jl | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/broadcast.jl b/test/broadcast.jl index 41ca604cb50e44..dee2fc4cc7554b 100644 --- a/test/broadcast.jl +++ b/test/broadcast.jl @@ -956,6 +956,10 @@ end @test sum(bc, dims=1, init=0) == [5] bc = Broadcast.instantiate(Broadcast.broadcasted(*, ['a','b'], 'c')) @test prod(bc, dims=1, init="") == ["acbc"] + + a = rand(-10:10,32,4); b = rand(-10:10,32,4) + bc = Broadcast.instantiate(Broadcast.broadcasted(+,a,b)) + @test sum(bc; dims = 1, init = 0.0) == sum(collect(bc); dims = 1, init = 0.0) end # treat Pair as scalar: