From 7b9a632bc26a8a6d8b015730c1b0f7bc29049b71 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Mon, 18 Mar 2024 16:59:33 +0100
Subject: [PATCH] feat: fixed test
---
unit_tests/recipes/Test_removeFlatten.cpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/unit_tests/recipes/Test_removeFlatten.cpp b/unit_tests/recipes/Test_removeFlatten.cpp
index 13e83f985..feb982fdf 100644
--- a/unit_tests/recipes/Test_removeFlatten.cpp
+++ b/unit_tests/recipes/Test_removeFlatten.cpp
@@ -31,7 +31,7 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
std::shared_ptr<Node> fc1 = FC(10, 10, "FC_2");
std::shared_ptr<Node> prod = Producer(std::array<DimSize_t, 10>(), "myProd");
- SECTION("flatten last layer") {
+ SECTION("flatten last layer : nothing removed because pattern searched is Flatten=>FC") {
std::shared_ptr<Aidge::GraphView> g = Sequential({fc0, flatten});
removeFlatten(g);
@@ -39,12 +39,12 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
- CHECK(g->getOrderedOutputs()[0].first == fc0);
+ CHECK(g->getOrderedOutputs()[0].first == flatten);
CHECK(fc0->getParent(0) == nullptr);
- CHECK(fc0->getChildren(0).size() == 0);
+ CHECK(fc0->getChildren(0).size() == 1);
CHECK(g->getRootNode() == fc0);
}
- SECTION("flatten first layer") {
+ SECTION("flatten first layer : flatten removed") {
auto g = Sequential({flatten, fc0});
removeFlatten(g);
--
GitLab