[WIP] [BlockSparseArrays] Redesign nested views

ITensor · Jun 19, 2024 · 4a22837 · 4a22837
1 parent 7b2ada9
commit 4a22837
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 147 deletions.
diff --git a/NDTensors/src/lib/BlockSparseArrays/test/backup/runtests.jl b/NDTensors/src/lib/BlockSparseArrays/test/backup/runtests.jl
diff --git a/NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl b/NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl
@@ -28,16 +28,23 @@ include("TestBlockSparseArraysUtils.jl")
 
     a = BlockSparseArray{elt}([2, 3], [3, 4])
     b = @view a[[Block(2), Block(1)], [Block(2), Block(1)]]
+    @test b isa SubArray{<:Any,<:Any,<:BlockSparseArray}
     @test_broken b[2:4, 2:4]
 
+    a = BlockSparseArray{elt}([2, 3], [3, 4])
+    b = @views a[[Block(2), Block(1)], [Block(2), Block(1)]][Block(1, 1)]
+    @test_broken b isa SubArray{<:Any,<:Any,<:BlockSparseArray}
+
     a = BlockSparseArray{elt}([2, 3], [3, 4])
     b = @views a[Block(1, 1)][1:2, 1:1]
+    @test b isa SubArray{<:Any,<:Any,<:BlockSparseArray}
     for i in parentindices(b)
       @test_broken i isa BlockSlice{<:BlockIndexRange{1}}
     end
 
     a = BlockSparseArray{elt}([2, 3], [3, 4])
     b = @view a[[Block(2), Block(1)], [Block(2), Block(1)]]
+    @test b isa SubArray{<:Any,<:Any,<:BlockSparseArray}
     @test_broken b[Block(1, 1)] = randn(3, 3)
   end
   @testset "Basics" begin