Add test to assign chunked array to chunked array

Author: davidbrochart

test/test_xchunk_store_manager.cpp

@@ -37,6 +37,17 @@ namespace xt
         }
     }
 
+    template <class T>
+    inline void check_file_equal(const std::string& path, const T&& ref)
+    {
+        std::ifstream in_file;
+        in_file.open(path);
+        xt::xarray<double> data;
+        data = xt::load_bin<double>(in_file);
+        EXPECT_TRUE(xt::all(equal(data, ref)));
+        in_file.close();
+    }
+
     TEST(xchunked_array, disk_array)
     {
         std::vector<size_t> shape = {4, 4};
@@ -121,4 +132,32 @@ namespace xt
             EXPECT_EQ(v, init_value);
         }
     }
+
+    TEST(xchunked_array, chunked_assign_chunked)
+    {
+        std::vector<size_t> shape = {4, 4};
+        std::vector<size_t> chunk_shape = {2, 2};
+        std::string chunk_dir1 = "files4";
+        fs::create_directory(chunk_dir1);
+        auto a1 = chunked_file_array<double, xio_disk_handler<xio_binary_config>>(shape, chunk_shape, chunk_dir1);
+        std::string chunk_dir2 = "files5";
+        fs::create_directory(chunk_dir2);
+        auto a2 = chunked_file_array<double, xio_disk_handler<xio_binary_config>>(shape, chunk_shape, chunk_dir2);
+        auto a3 = arange(4 * 4).reshape({4, 4});
+        a2 = a3;
+        a2.chunks().flush();
+        // check that a2 has correct chunks
+        check_file_equal(chunk_dir2 + "/0.0", xt::xarray<double>({{0, 1}, {4, 5}}));
+        check_file_equal(chunk_dir2 + "/1.0", xt::xarray<double>({{8, 9}, {12, 13}}));
+        check_file_equal(chunk_dir2 + "/0.1", xt::xarray<double>({{2, 3}, {6, 7}}));
+        check_file_equal(chunk_dir2 + "/1.1", xt::xarray<double>({{10, 11}, {14, 15}}));
+
+        a1 = a2;
+        a1.chunks().flush();
+        // check that a1 has correct chunks
+        check_file_equal(chunk_dir1 + "/0.0", xt::xarray<double>({{0, 1}, {4, 5}}));
+        check_file_equal(chunk_dir1 + "/1.0", xt::xarray<double>({{8, 9}, {12, 13}}));
+        check_file_equal(chunk_dir1 + "/0.1", xt::xarray<double>({{2, 3}, {6, 7}}));
+        check_file_equal(chunk_dir1 + "/1.1", xt::xarray<double>({{10, 11}, {14, 15}}));
+    }
 }