Test MatchingPartitioner in atlas_test_functionspace_splitcomm

ecmwf · Aug 30, 2023 · 2ba3728 · 2ba3728
1 parent b5f8bda
commit 2ba3728
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/tests/functionspace/test_functionspace_splitcomm.cc b/src/tests/functionspace/test_functionspace_splitcomm.cc
@@ -15,6 +15,7 @@
 #include "tests/AtlasTestEnvironment.h"
 #include "atlas/functionspace/NodeColumns.h"
 #include "atlas/functionspace/StructuredColumns.h"
+#include "atlas/grid/Partitioner.h"
 #include "atlas/field/for_each.h"
 
 namespace atlas {
@@ -142,6 +143,17 @@ CASE("test FunctionSpace StructuredColumns") {
 
 //-----------------------------------------------------------------------------
 
+CASE("test FunctionSpace StructuredColumns with MatchingPartitioner") {
+    Fixture fixture;
+
+    auto fs_A = functionspace::StructuredColumns(grid(), option::mpi_split_comm());
+    auto fs_B = functionspace::StructuredColumns(grid(), grid::MatchingPartitioner(fs_A), option::mpi_split_comm());
+    fs_A.polygon().outputPythonScript("fs_A_polygons.py");
+    fs_B.polygon().outputPythonScript("fs_B_polygons.py");
+}
+
+//-----------------------------------------------------------------------------
+
 }  // namespace test
 }  // namespace atlas