diff --git a/unit_tests/data/Test_TensorImpl.cpp b/unit_tests/data/Test_TensorImpl.cpp index e2a8840b4a9179c316d3fd2eb06197c25c6bb09d..9d8bbb2e727dcb6eff1765134dc841d4241db728 100644 --- a/unit_tests/data/Test_TensorImpl.cpp +++ b/unit_tests/data/Test_TensorImpl.cpp @@ -193,18 +193,18 @@ TEST_CASE("Tensor extract") Rainbow.setDatatype(DataType::UInt16); Rainbow.setBackend("cpu"); MakeRainbow<std::uint16_t>(Rainbow); - // Tensor view(Rainbow, {2, 2, 3}, {0, 1, 1}); - // for (Coord_t a = 0; a < view.dims()[0]; ++a) - // { - // for (Coord_t b = 0; b < view.dims()[1]; ++b) - // { - // for (Coord_t c = 0; c < view.dims()[2]; ++c) - // { - // REQUIRE( - // view.get<std::uint16_t>({a, b + 1, c + 1}) - // == Rainbow.get<std::uint16_t>({a, b + 1, c + 1})); - // } - // } - // } + Tensor view(Rainbow, {2, 2, 3}, {0, 1, 1}); + for (Coord_t a = 0; a < view.dims()[0]; ++a) + { + for (Coord_t b = 0; b < view.dims()[1]; ++b) + { + for (Coord_t c = 0; c < view.dims()[2]; ++c) + { + REQUIRE( + view.get<std::uint16_t>({a, b + 1, c + 1}) + == Rainbow.get<std::uint16_t>({a, b + 1, c + 1})); + } + } + } } } \ No newline at end of file