Convert print statements to asserts.

BlueBrain · Mar 2, 2023 · cfd9735 · cfd9735
1 parent fa0d296
commit cfd9735
Showing 1 changed file with 19 additions and 14 deletions.
diff --git a/tests/unit/tests_high_five_base.cpp b/tests/unit/tests_high_five_base.cpp
@@ -1740,13 +1740,11 @@ TEST_CASE("ReadInBroadcastDims") {
     // Create the data space for the dataset.
     std::vector<size_t> dims_a{1, 3};
     std::vector<size_t> dims_b{1, 3, 1};
-    std::vector<size_t> dims_c{3, 2, 1};
+    std::vector<size_t> dims_c{2, 3, 1};
 
     // broadcastable input / output vectors
     std::vector<double> some_data{5.0, 6.0, 7.0};
-    std::vector<double> other_data{-5.0, -6.0};
-    std::vector<std::vector<double>> some_data_2d{some_data};
-    std::vector<std::vector<double>> other_data_2d{other_data, other_data, other_data};
+    std::vector<std::vector<double>> some_data_2d{some_data, some_data};
 
     std::vector<double> data_a;
 
@@ -1785,18 +1783,25 @@ TEST_CASE("ReadInBroadcastDims") {
 
     CHECK(data_a == some_data);
 
-    for (size_t i = 0; i < some_data.size(); ++i) {
-        CHECK(data_b_2d[0][i] == some_data[i]);
-    }
+    CHECK(data_b_1d == some_data);
+
+    CHECK(data_b_2d.size() == 1);
+    CHECK(data_b_2d[0] == some_data);
+
+    CHECK(data_b_3d.size() == 1);
+    CHECK(data_b_3d[0].size() == 1);
+    CHECK(data_b_3d[0][0] == some_data);
 
-    std::cout << "data_b_1d.shape = " << data_b_1d.size() << "\n";
-    std::cout << "data_b_2d.shape = " << data_b_2d.size() << ", " << data_b_2d[0].size() << "\n";
-    std::cout << "data_b_3d.shape = " << data_b_3d.size() << ", " << data_b_3d[0].size() << ", "
-              << data_b_3d[0][0].size() << "\n";
+    CHECK(data_c_2d == some_data_2d);
 
-    std::cout << "data_c_2d.shape = " << data_c_2d.size() << ", " << data_c_2d[0].size() << "\n";
-    std::cout << "data_c_3d.shape = " << data_c_3d.size() << ", " << data_c_3d[0].size() << ", "
-              << data_c_3d[0][0].size() << "\n";
+    CHECK(data_c_3d.size() == dims_c[0]);
+    for (size_t i = 0; i < dims_c[0]; ++i) {
+        CHECK(data_c_3d[i].size() == dims_c[1]);
+        for (size_t j = 0; j < dims_c[1]; ++j) {
+            CHECK(data_c_3d[i][j].size() == 1ul);
+            CHECK(data_c_3d[i][j][0] = some_data_2d[i][j]);
+        }
+    }
 }
 
 TEST_CASE("HighFiveRecursiveGroups") {