Fix for infinite loop when passing 0d array to setindex of n-dim arra…

…ys (JuliaLang#39608) * Fix for infinite loop with 0d array * Consistency * More consistency * Test
ElOceanografo · May 4, 2021 · a3fb768 · a3fb768
1 parent 8e19adc
commit a3fb768
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/base/indices.jl b/base/indices.jl
@@ -239,6 +239,9 @@ setindex_shape_check(X::AbstractArray) =
 setindex_shape_check(X::AbstractArray, i::Integer) =
     (length(X)==i || throw_setindex_mismatch(X, (i,)))
 
+setindex_shape_check(X::AbstractArray{<:Any, 0}, i::Integer...) =
+    (length(X) == prod(i) || throw_setindex_mismatch(X, i))
+
 setindex_shape_check(X::AbstractArray{<:Any,1}, i::Integer) =
     (length(X)==i || throw_setindex_mismatch(X, (i,)))
 

diff --git a/test/arrayops.jl b/test/arrayops.jl
@@ -2898,3 +2898,9 @@ end
     @test as isa TSlow{Int,3}
     @test size(as) == (3, 5, 1)
 end
+
+@testset "0-dimensional shape checking #39608" begin
+    @test [fill(1); [2; 2]] == [1; 2; 2]
+    @test [fill(1); fill(2, (2,1,1))] == reshape([1; 2; 2], (3, 1, 1))
+    @test_throws DimensionMismatch [fill(1); rand(2, 2, 2)]
+end