1 files changed, 101 insertions, 0 deletions
diff --git a/compiler/moco/import/src/TestHelper.test.cpp b/compiler/moco/import/src/TestHelper.test.cpp
new file mode 100644
index 000000000..06c3dd372
--- /dev/null
+++ b/compiler/moco/import/src/TestHelper.test.cpp
@@ -0,0 +1,101 @@
+/*
+ * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "TestHelper.h"
+
+#include <moco/IR/Nodes/TFConst.h>
+#include <stdex/Memory.h>
+
+#include <gtest/gtest.h>
+
+namespace moco
+{
+namespace test
+{
+
+TFNodeBuildTester::TFNodeBuildTester()
+{
+  _graph = loco::make_graph();
+  _tensor_names = stdex::make_unique<moco::SymbolTable>();
+}
+
+void TFNodeBuildTester::inputs(const std::vector<std::string> &names)
+{
+  for (auto name : names)
+  {
+    auto input = _graph->nodes()->create<moco::TFConst>();
+    moco::TensorName name_01(name, 0);
+    _tensor_names->enroll(name_01, input);
+
+    _inputs.push_back(input);
+  }
+}
+
+void TFNodeBuildTester::inputs(const std::vector<std::string> &names, const loco::DataType dtype)
+{
+  for (auto name : names)
+  {
+    auto input = _graph->nodes()->create<moco::TFConst>();
+    input->dtype(dtype);
+    moco::TensorName name_01(name, 0);
+    _tensor_names->enroll(name_01, input);
+
+    _inputs.push_back(input);
+  }
+}
+
+void TFNodeBuildTester::output(const char *name) { _output = name; }
+
+moco::TFNode *TFNodeBuildTester::output(void)
+{
+  assert(_output != nullptr);
+
+  moco::TensorName tname(_output, 0);
+  return static_cast<moco::TFNode *>(_tensor_names->node(tname));
+}
+
+void TFNodeBuildTester::run(tensorflow::NodeDef &nodedef, moco::GraphBuilder &graphbuilder)
+{
+  assert(_output != nullptr);
+
+  auto node_defs = stdex::make_unique<moco::NodeDefTable>();
+  auto updates = stdex::make_unique<moco::UpdateQueue>();
+
+  moco::GraphBuilderContext gb_context(_graph.get(), node_defs.get(), _tensor_names.get(),
+                                       updates.get());
+
+  EXPECT_TRUE(graphbuilder.validate(nodedef));
+  graphbuilder.build(nodedef, &gb_context);
+
+  for (auto &update : updates->queue())
+  {
+    update->input(_tensor_names.get());
+  }
+
+  auto tfnode = output();
+  ASSERT_NE(tfnode, nullptr);
+  ASSERT_STREQ(tfnode->name().c_str(), _output);
+
+  int idx = 0;
+  ASSERT_EQ(tfnode->arity(), _inputs.size());
+  for (auto input : _inputs)
+  {
+    ASSERT_EQ(tfnode->arg(idx++), input);
+  }
+}
+
+} // namespace test
+} // namespace moco