Fixes in tests

gridap · Mar 21, 2024 · ae298c8 · ae298c8
1 parent 40a7fd2
commit ae298c8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 10 deletions.
diff --git a/src/Algebra.jl b/src/Algebra.jl
@@ -169,9 +169,11 @@ function i_am_in(comm::MPIVoidVector)
 end
 
 function change_parts(x::Union{MPIArray,DebugArray,Nothing,MPIVoidVector}, new_parts; default=nothing)
-  x_new = map(new_parts) do _p
-    if isa(x,MPIArray) || isa(x,DebugArray)
+  x_new = map(new_parts) do p
+    if isa(x,MPIArray)
       PartitionedArrays.getany(x)
+    elseif isa(x,DebugArray) && (p <= length(x.items))
+      x.items[p]
     else
       default
     end

diff --git a/test/PLaplacianTests.jl b/test/PLaplacianTests.jl
@@ -7,13 +7,6 @@ using PartitionedArrays
 using Test
 using SparseArrays
 
-# Overload required for the tests below
-function Base.copy(a::PSparseMatrix)
-  a_matrix_partition = similar(a.matrix_partition)
-  copy!(a_matrix_partition, a.matrix_partition)
-  PSparseMatrix(a_matrix_partition,a.row_partition,a.col_partition)
-end
-
 function main(distribute,parts)
   main(distribute,parts,FullyAssembledRows(),SparseMatrixCSR{0,Float64,Int})
   main(distribute,parts,SubAssembledRows(),SparseMatrixCSC{Float64,Int})

diff --git a/test/StokesOpenBoundaryTests.jl b/test/StokesOpenBoundaryTests.jl
@@ -3,7 +3,7 @@ module StokesOpenBoundaryTests
 using Gridap
 using LinearAlgebra
 using Test
-using Gridap.ODEs
+using Gridap.ODEs, Gridap.Algebra
 using GridapDistributed
 using PartitionedArrays