Cleaning up unit tests

Cargo.toml

@@ -30,7 +30,8 @@ itertools = "0.13.*"
 mpi = { version = "0.8.*"}
 num = "0.4"
 ndelement = { git="https://github.com/bempp/ndelement.git", features = ["mpi"]}
-ndgrid = { git="https://github.com/bempp/ndgrid.git", features = ["mpi"] }
+#ndgrid = { git="https://github.com/bempp/ndgrid.git", features = ["mpi"] }
+ndgrid = { path = "../ndgrid" }
 rayon = "1.9"
 rlst = { git = "https://github.com/linalg-rs/rlst.git", features = ["mpi"] }
 green-kernels = { git = "https://github.com/bempp/green-kernels.git", features = ["mpi"] }

examples/test_parallel_assembly.rs

@@ -18,9 +18,8 @@ use ndelement::{
     types::{Continuity, ReferenceCellType},
 };
 use ndgrid::{
-    grid::parallel::ParallelGrid,
     traits::{Builder, Entity, Grid, ParallelBuilder},
-    SingleElementGrid, SingleElementGridBuilder,
+    ParallelGrid, SingleElementGrid, SingleElementGridBuilder,
 };
 use rlst::{CsrMatrix, Shape};
 use std::collections::{hash_map::Entry, HashMap};
@@ -55,9 +54,9 @@ fn example_single_element_grid<C: Communicator>(
 
     if rank == 0 {
         create_single_element_grid_data(&mut b, n);
-        b.create_parallel_grid(comm)
+        b.create_parallel_grid_root(comm)
     } else {
-        b.receive_parallel_grid(comm, 0)
+        b.create_parallel_grid(comm, 0)
     }
 }
 

src/shapes.rs

@@ -106,9 +106,9 @@ pub fn regular_sphere<T: RealScalar + Equivalence, C: Communicator>(
             b.add_cell(i, v);
         }
 
-        b.create_parallel_grid(comm)
+        b.create_parallel_grid_root(comm)
     } else {
         SingleElementGridBuilder::new(3, (ReferenceCellType::Triangle, degree))
-            .receive_parallel_grid(comm, 0)
+            .create_parallel_grid(comm, 0)
     }
 }