Test getindex

Author: janEbert

test/sparse.jl

@@ -18,6 +18,15 @@ blockdim = 5
     @test size(d_x,1) == m
     @test size(d_x,2) == 1
     @test ndims(d_x)  == 1
+    CuArrays.@allowscalar begin
+        @test Array(d_x[:])        == x[:]
+        @test d_x[firstindex(d_x)] == x[firstindex(x)]
+        @test d_x[div(end, 2)]     == x[div(end, 2)]
+        @test d_x[end]             == x[end]
+        @test Array(d_x[firstindex(d_x):end]) == x[firstindex(x):end]
+    end
+    @test_throws BoundsError d_x[firstindex(d_x) - 1]
+    @test_throws BoundsError d_x[end + 1]
     @test nnz(d_x)    == nnz(x)
     @test Array(nonzeros(d_x)) == nonzeros(x)
     @test Array(SparseArrays.nonzeroinds(d_x)) == SparseArrays.nonzeroinds(x)
@@ -30,6 +39,26 @@ blockdim = 5
     @test size(d_x,2) == n
     @test size(d_x,3) == 1
     @test ndims(d_x)  == 2
+    CuArrays.@allowscalar begin
+        @test Array(d_x[:])        == x[:]
+        @test d_x[firstindex(d_x)] == x[firstindex(x)]
+        @test d_x[div(end, 2)]     == x[div(end, 2)]
+        @test d_x[end]             == x[end]
+        @test d_x[firstindex(d_x), firstindex(d_x)] == x[firstindex(x), firstindex(x)]
+        @test d_x[div(end, 2), div(end, 2)]         == x[div(end, 2), div(end, 2)]
+        @test d_x[end, end]        == x[end, end]
+        @test Array(d_x[firstindex(d_x):end, firstindex(d_x):end]) == x[:, :]
+    end
+    @test_throws BoundsError d_x[firstindex(d_x) - 1]
+    @test_throws BoundsError d_x[end + 1]
+    @test_throws BoundsError d_x[firstindex(d_x) - 1, firstindex(d_x) - 1]
+    @test_throws BoundsError d_x[end + 1, end + 1]
+    @test_throws BoundsError d_x[firstindex(d_x) - 1:end + 1, :]
+    @test_throws BoundsError d_x[firstindex(d_x) - 1, :]
+    @test_throws BoundsError d_x[end + 1, :]
+    @test_throws BoundsError d_x[:, firstindex(d_x) - 1:end + 1]
+    @test_throws BoundsError d_x[:, firstindex(d_x) - 1]
+    @test_throws BoundsError d_x[:, end + 1]
     @test nnz(d_x)    == nnz(x)
     @test Array(nonzeros(d_x)) == nonzeros(x)
     @test nnz(d_x)    == length(nonzeros(d_x))