[WIP] Add C++ test

Signed-off-by: take-cheeze <[email protected]>

Author: take-cheeze

CMakeLists.txt

@@ -98,3 +98,10 @@ install(TARGETS
   onnx_optimizer onnx_optimizer_c_api
   EXPORT ONNXOptimizerTargets DESTINATION ${CMAKE_INSTALL_LIBDIR})
 
+if(ONNX_BUILD_TESTS)
+  enable_testing()
+  include(third_party/onnx/cmake/external/googletest.cmake)
+  add_executable(test_simple tests/test_simple.cc)
+  target_link_libraries(test_simple gtest)
+  add_test(NAME test_simple COMMAND ${CMAKE_CURRENT_BINARY_DIR}/test_simple)
+endif()

tests/test_simple.cc

@@ -0,0 +1,16 @@
+#include <gtest/gtest.h>
+
+TEST(OptimizerTest, NopReshape) {
+    Graph graph = parser.parse_graph(R"(
+        agraph (float[5, 7] X) => (float[5, 7] Z)
+        {
+            Shape = Constant<value=int64[2]{5, -1}> ()
+            Y = Reshape (X, Shape)
+            Z = Identity(Y)
+        }
+    )");
+    OptimizedModel optimized_model = _optimized(graph, {"eliminate_nop_reshape", "eliminate_deadend"}, false);
+
+    ASSERT_EQ(optimized_model.graph.node.size(), 1);
+    ASSERT_EQ(optimized_model.graph.node[0].op_type, "Identity");
+}