From 7fdd6b9377df36b5de6616a14f336a501f5aebaa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Franz=20P=C3=B6schel?= <franz.poeschel@gmail.com>
Date: Tue, 15 Dec 2020 13:36:44 +0100
Subject: [PATCH] Add parallel test

---
 test/ParallelIOTest.cpp | 73 ++++++++++++++++++++++++++++++++++++-----
 1 file changed, 64 insertions(+), 9 deletions(-)

diff --git a/test/ParallelIOTest.cpp b/test/ParallelIOTest.cpp
index 4460c019eb..9c5df05652 100644
--- a/test/ParallelIOTest.cpp
+++ b/test/ParallelIOTest.cpp
@@ -7,15 +7,15 @@
 #include <catch2/catch.hpp>
 
 #if openPMD_HAVE_MPI
-#   include <mpi.h>
-
-#   include <iostream>
-#   include <algorithm>
-#   include <string>
-#   include <vector>
-#   include <list>
-#   include <memory>
-#   include <tuple>
+#    include <algorithm>
+#    include <iostream>
+#    include <list>
+#    include <memory>
+#    include <mpi.h>
+#    include <numeric> // std::iota
+#    include <string>
+#    include <tuple>
+#    include <vector>
 
 using namespace openPMD;
 
@@ -337,6 +337,61 @@ TEST_CASE( "available_chunks_test", "[parallel][adios]" )
     available_chunks_test( "bp" );
 }
 
+void
+extendDataset( std::string const & ext )
+{
+    std::string filename = "../samples/parallelExtendDataset." + ext;
+    int r_mpi_rank{ -1 }, r_mpi_size{ -1 };
+    MPI_Comm_rank( MPI_COMM_WORLD, &r_mpi_rank );
+    MPI_Comm_size( MPI_COMM_WORLD, &r_mpi_size );
+    unsigned mpi_rank{ static_cast< unsigned >( r_mpi_rank ) },
+        mpi_size{ static_cast< unsigned >( r_mpi_size ) };
+    std::vector< int > data1( 25 );
+    std::vector< int > data2( 25 );
+    std::iota( data1.begin(), data1.end(), 0 );
+    std::iota( data2.begin(), data2.end(), 25 );
+    {
+        Series write( filename, Access::CREATE, MPI_COMM_WORLD );
+        if( ext == "bp" && write.backend() != "ADIOS2" )
+        {
+            // dataset resizing unsupported in ADIOS1
+            return;
+        }
+        Dataset ds1{ Datatype::INT, { mpi_size, 25 } };
+        Dataset ds2{ { mpi_size, 50 } };
+
+        // array record component -> array record component
+        // should work
+        auto E_x = write.iterations[ 0 ].meshes[ "E" ][ "x" ];
+        E_x.resetDataset( ds1 );
+        E_x.storeChunk( data1, { mpi_rank, 0 }, { 1, 25 } );
+        write.flush();
+
+        E_x.resetDataset( ds2 );
+        E_x.storeChunk( data2, { mpi_rank, 25 }, { 1, 25 } );
+        write.flush();
+    }
+
+    {
+        Series read( filename, Access::READ_ONLY );
+        auto E_x = read.iterations[ 0 ].meshes[ "E" ][ "x" ];
+        REQUIRE( E_x.getExtent() == Extent{ mpi_size, 50 } );
+        auto chunk = E_x.loadChunk< int >( { 0, 0 }, { mpi_size, 50 } );
+        read.flush();
+        for( size_t rank = 0; rank < mpi_size; ++rank )
+        {
+            for( size_t i = 0; i < 50; ++i )
+            {
+                REQUIRE( chunk.get()[ i ] == i );
+            }
+        }
+    }
+}
+
+TEST_CASE( "extend_dataset", "[parallel]" )
+{
+    extendDataset( "bp" );
+}
 #endif
 
 #if openPMD_HAVE_ADIOS1 && openPMD_HAVE_MPI