diff options
Diffstat (limited to 'runtimes/neurun/test/graph/operation/Set.cc')
-rw-r--r-- | runtimes/neurun/test/graph/operation/Set.cc | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/runtimes/neurun/test/graph/operation/Set.cc b/runtimes/neurun/test/graph/operation/Set.cc index 3560482ee..3c5fd8345 100644 --- a/runtimes/neurun/test/graph/operation/Set.cc +++ b/runtimes/neurun/test/graph/operation/Set.cc @@ -17,18 +17,18 @@ #include <gtest/gtest.h> #include "../MockNode.h" -#include "model/operation/Set.h" +#include "model/Operations.h" -using neurun::model::operation::Set; -using neurun::model::operation::Node; -using neurun::model::operation::Index; +using neurun::model::Operations; +using neurun::model::Operation; +using neurun::model::OperationIndex; TEST(graph_operation_Set, operation_test) { - Set set; - set.append( - std::unique_ptr<Node>(new neurun_test::graph::SimpleMockNode({1, 2, 3, 4}, {5, 6, 7}))); - Index idx{0u}; - ASSERT_EQ(set.at(idx).getInputs().size(), 4); - ASSERT_EQ(set.at(idx).getOutputs().size(), 3); + Operations ops; + ops.push( + std::unique_ptr<Operation>(new neurun_test::graph::SimpleMockNode({1, 2, 3, 4}, {5, 6, 7}))); + OperationIndex idx{0u}; + ASSERT_EQ(ops.at(idx).getInputs().size(), 4); + ASSERT_EQ(ops.at(idx).getOutputs().size(), 3); } |