diff --git a/unit_tests/recipes/Test_removeFlatten.cpp b/unit_tests/recipes/Test_removeFlatten.cpp index 572fd121a277731f457e04fc54aa63eef24af2c4..84157f0c21aef2a775011e3ccd130511252e94e9 100644 --- a/unit_tests/recipes/Test_removeFlatten.cpp +++ b/unit_tests/recipes/Test_removeFlatten.cpp @@ -34,9 +34,8 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { SECTION("flatten last layer : nothing removed because pattern searched is " "Flatten=>FC") { std::shared_ptr<Aidge::GraphView> g = Sequential({fc0, flatten}); - fmt::print("debug {}", __LINE__); + removeFlatten(g); - fmt::print("debug {}", __LINE__); CHECK(g->getOrderedInputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1); @@ -48,10 +47,8 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { } SECTION("flatten first layer : flatten removed") { auto g = Sequential({flatten, fc0}); - fmt::print("debug {}", __LINE__); removeFlatten(g); - fmt::print("debug {}", __LINE__); CHECK(g->getOrderedInputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1); @@ -62,11 +59,10 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { CHECK(g->getRootNode() == fc0); } SECTION("flatten middle layer") { + auto g = Sequential({fc0, flatten, fc1}); - fmt::print("debug {}", __LINE__); removeFlatten(g); - fmt::print("debug {}", __LINE__); CHECK(g->getOrderedInputs().size() == 1); CHECK(g->getOrderedOutputs().size() == 1); @@ -81,11 +77,9 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") { // prod->addChild(flatten, 0); // flatten->addChild(fc0, 0); // auto g = std::make_shared<GraphView>({prod, flatten, fc0}); - fmt::print("debug {}", __LINE__); removeFlatten(g); - fmt::print("debug {}", __LINE__); CHECK(g->getOrderedInputs().size() == 0); CHECK(g->getOrderedOutputs().size() == 1); CHECK(g->getOrderedInputs()[0].first == fc0);