Skip to content

Commit fd5b611

Browse files
authored
Test BlockSparseArrays broadcasting (#47)
1 parent 4c38a99 commit fd5b611

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

test/test_blockarraysext.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ using Test: @test, @testset
55

66
@testset "NamedDimsArraysBlockArraysExt" begin
77
elt = Float64
8+
89
a = BlockSparseArray{elt}([2, 3], [2, 3])
910
a[Block(2, 1)] = randn(elt, 3, 2)
1011
a[Block(1, 2)] = randn(elt, 2, 3)
@@ -18,4 +19,16 @@ using Test: @test, @testset
1819
@test dename(n[i[Block.(1:2)], j[Block(1)]]) == a[Block.(1:2), Block(1)]
1920
@test dename(n[Block.(1:2), Block(1)]) == a[Block.(1:2), Block(1)]
2021
@test dename(n[Block.(1:2), Block.(1:2)]) == a[Block.(1:2), Block.(1:2)]
22+
23+
a = BlockSparseArray{elt}([2, 3], [2, 3])
24+
a[Block(2, 1)] = randn(elt, 3, 2)
25+
a[Block(1, 2)] = randn(elt, 2, 3)
26+
b = BlockSparseArray{elt}([2, 3], [2, 3])
27+
b[Block(2, 1)] = randn(elt, 3, 2)
28+
b[Block(1, 2)] = randn(elt, 2, 3)
29+
na = nameddimsarray(a, ("i", "j"))
30+
nb = nameddimsarray(b, ("j", "i"))
31+
nc = na .+ 2 .* nb
32+
c = a + 2 * permutedims(b, (2, 1))
33+
@test dename(nc, ("i", "j")) c
2134
end

0 commit comments

Comments
 (0)