Skip to content
Snippets Groups Projects
Commit 646b1cb5 authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

chore : reordered tests

parent 59b4f17d
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
......@@ -22,7 +22,6 @@
namespace Aidge {
void removeFlatten(std::shared_ptr<Node> flatten) {
GraphView::replace({flatten}, {});
}
......
......@@ -38,10 +38,12 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
removeFlatten(g);
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
CHECK(g->getOrderedOutputs()[0].first == flatten);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
CHECK(fc0->getParent(0) == nullptr);
CHECK(fc0->getChildren(0).size() == 1);
CHECK(g->getRootNode() == fc0);
......@@ -54,9 +56,11 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedOutputs()[0].first == fc0);
CHECK(fc0->getParent(0) == nullptr);
CHECK(fc0->getChildren(0).size() == 0);
CHECK(g->getRootNode() == fc0);
......@@ -69,9 +73,11 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 1);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedOutputs()[0].first == fc1);
CHECK(fc1->getParent(0) == fc0);
CHECK(fc0->getChildren(0)[0] == fc1);
CHECK(g->getRootNode() == fc0);
......@@ -87,11 +93,13 @@ TEST_CASE("[cpu/recipies] RemoveFlatten", "[RemoveFlatten][recipies]") {
fmt::print("debug {}", __LINE__);
CHECK(g->getOrderedInputs().size() == 0);
CHECK(g->getOrderedOutputs().size() == 1);
CHECK(g->getOrderedInputs()[0].first == fc0);
CHECK(g->getOrderedOutputs()[0].first == fc0);
CHECK(fc0->getParent(0) == prod);
CHECK(fc0->getChildren(0).size() == 0);
CHECK(g->getRootNode() == prod);
}
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment