|
166 | 166 | end |
167 | 167 | end |
168 | 168 | end |
| 169 | + |
| 170 | +# Test that concatenations of annotated sparse/special matrix types with other matrix |
| 171 | +# types yield sparse arrays, and that the code which effects that does not make concatenations |
| 172 | +# strictly involving un/annotated dense matrices yield sparse arrays |
| 173 | +# |
| 174 | +# TODO: As with the associated code, these tests should be moved to a more appropriate |
| 175 | +# location, particularly some future equivalent of base/linalg/special.jl dedicated to |
| 176 | +# intereactions between a broader set of matrix types |
| 177 | +let |
| 178 | + N = 4 |
| 179 | + # The tested annotation types |
| 180 | + testfull = Bool(parse(Int,(get(ENV, "JULIA_TESTFULL", "0")))) |
| 181 | + utriannotations = (UpperTriangular, Base.LinAlg.UnitUpperTriangular) |
| 182 | + ltriannotations = (LowerTriangular, Base.LinAlg.UnitLowerTriangular) |
| 183 | + triannotations = (utriannotations..., ltriannotations...) |
| 184 | + symannotations = (Symmetric, Hermitian) |
| 185 | + annotations = testfull ? (triannotations..., symannotations...) : (LowerTriangular, Symmetric) |
| 186 | + # Concatenations involving these types, un/annotated, should yield sparse arrays |
| 187 | + spvec = spzeros(N) |
| 188 | + spmat = speye(N) |
| 189 | + diagmat = Diagonal(ones(N)) |
| 190 | + bidiagmat = Bidiagonal(ones(N), ones(N-1), true) |
| 191 | + tridiagmat = Tridiagonal(ones(N-1), ones(N), ones(N-1)) |
| 192 | + symtridiagmat = SymTridiagonal(ones(N), ones(N-1)) |
| 193 | + sparseconcatmats = testfull ? (spmat, diagmat, bidiagmat, tridiagmat, symtridiagmat) : (spmat, diagmat) |
| 194 | + # Concatenations involving strictly these types, un/annotated, should yield dense arrays |
| 195 | + densevec = ones(N) |
| 196 | + densemat = ones(N, N) |
| 197 | + # Annotated collections |
| 198 | + annodmats = [annot(densemat) for annot in annotations] |
| 199 | + annospcmats = [annot(spcmat) for annot in annotations, spcmat in sparseconcatmats] |
| 200 | + # Test that concatenations of pairwise combinations of annotated sparse/special |
| 201 | + # yield sparse matrices |
| 202 | + for annospcmata in annospcmats, annospcmatb in annospcmats |
| 203 | + @test issparse(vcat(annospcmata, annospcmatb)) |
| 204 | + @test issparse(hcat(annospcmata, annospcmatb)) |
| 205 | + @test issparse(hvcat((2,), annospcmata, annospcmatb)) |
| 206 | + @test issparse(cat((1,2), annospcmata, annospcmatb)) |
| 207 | + end |
| 208 | + # Test that concatenations of pairwise combinations of annotated sparse/special |
| 209 | + # matrices and other matrix/vector types yield sparse matrices |
| 210 | + for annospcmat in annospcmats |
| 211 | + # --> Tests applicable to pairs including only matrices |
| 212 | + for othermat in (densemat, annodmats..., sparseconcatmats...) |
| 213 | + @test issparse(vcat(annospcmat, othermat)) |
| 214 | + @test issparse(vcat(othermat, annospcmat)) |
| 215 | + end |
| 216 | + # --> Tests applicable to pairs including other vectors or matrices |
| 217 | + for other in (spvec, densevec, densemat, annodmats..., sparseconcatmats...) |
| 218 | + @test issparse(hcat(annospcmat, other)) |
| 219 | + @test issparse(hcat(other, annospcmat)) |
| 220 | + @test issparse(hvcat((2,), annospcmat, other)) |
| 221 | + @test issparse(hvcat((2,), other, annospcmat)) |
| 222 | + @test issparse(cat((1,2), annospcmat, other)) |
| 223 | + @test issparse(cat((1,2), other, annospcmat)) |
| 224 | + end |
| 225 | + end |
| 226 | + # The preceding tests should cover multi-way combinations of those types, but for good |
| 227 | + # measure test a few multi-way combinations involving those types |
| 228 | + @test issparse(vcat(spmat, densemat, annospcmats[1], annodmats[2])) |
| 229 | + @test issparse(vcat(densemat, spmat, annodmats[1], annospcmats[2])) |
| 230 | + @test issparse(hcat(spvec, annodmats[1], annospcmats[3], densevec, diagmat)) |
| 231 | + @test issparse(hcat(annodmats[2], annospcmats[4], spvec, densevec, diagmat)) |
| 232 | + @test issparse(hvcat((5,), diagmat, densevec, spvec, annodmats[1], annospcmats[1])) |
| 233 | + @test issparse(hvcat((5,), spvec, annodmats[2], diagmat, densevec, annospcmats[2])) |
| 234 | + @test issparse(cat((1,2), annodmats[1], diagmat, annospcmats[3], densevec, spvec)) |
| 235 | + @test issparse(cat((1,2), spvec, diagmat, densevec, annospcmats[4], annodmats[2])) |
| 236 | + # Test that concatenations strictly involving un/annotated dense matrices/vectors |
| 237 | + # yield dense arrays |
| 238 | + for densemata in (densemat, annodmats...) |
| 239 | + # --> Tests applicable to pairs including only matrices |
| 240 | + for densematb in (densemat, annodmats...) |
| 241 | + @test !issparse(vcat(densemata, densematb)) |
| 242 | + @test !issparse(vcat(densematb, densemata)) |
| 243 | + end |
| 244 | + # --> Tests applicable to pairs including vectors or matrices |
| 245 | + for otherdense in (densevec, densemat, annodmats...) |
| 246 | + @test !issparse(hcat(densemata, otherdense)) |
| 247 | + @test !issparse(hcat(otherdense, densemata)) |
| 248 | + @test !issparse(hvcat((2,), densemata, otherdense)) |
| 249 | + @test !issparse(hvcat((2,), otherdense, densemata)) |
| 250 | + @test !issparse(cat((1,2), densemata, otherdense)) |
| 251 | + @test !issparse(cat((1,2), otherdense, densemata)) |
| 252 | + end |
| 253 | + end |
| 254 | +end |
0 commit comments