diff --git a/unit_tests/operator/Test_ReduceMeanImpl.cpp b/unit_tests/operator/Test_ReduceMeanImpl.cpp index 6a5166c7b22784554f86c9b66c213f321eec4fc6..2cde38d690a53b0bb3fb74d1ff8ff5f1b7a0c8fd 100644 --- a/unit_tests/operator/Test_ReduceMeanImpl.cpp +++ b/unit_tests/operator/Test_ReduceMeanImpl.cpp @@ -20,7 +20,7 @@ using namespace Aidge; -TEST_CASE("[cpu/operator] ReduceMean(forward)") { +TEST_CASE("[cpu/operator] ReduceMean(forward)", "[ReduceMean][CPU]") { SECTION("KeepDims") { std::shared_ptr<Tensor> myInput = std::make_shared<Tensor>(Array3D<float,3,2,2> { { @@ -76,7 +76,7 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)") { } }); std::shared_ptr<Tensor> myOutput = std::make_shared<Tensor>(Array2D<float,3,2> { - { + { { 12.5, 1.5 }, { 35.0, 1.5 }, { 57.5, 1.5 } @@ -124,7 +124,7 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)") { op->computeOutputDims(); myReduceMean->forward(); op->getOutput(0)->print(); - + REQUIRE(*(op->getOutput(0)) == *myOutput); } } \ No newline at end of file