[FRONT-6] add tests

CMakeLists.txt

@@ -8,6 +8,8 @@ set(CMAKE_CXX_STANDARD_REQUIRED ON)
 find_package(Protobuf REQUIRED)
 find_package(absl REQUIRED)
 
+add_compile_definitions(PROJECT_ROOT="${CMAKE_CURRENT_SOURCE_DIR}")
+
 set(ONNX_PROTO "${CMAKE_CURRENT_SOURCE_DIR}/onnx.proto")
 
 protobuf_generate_cpp(
@@ -73,5 +75,5 @@ target_link_libraries(run
 
 add_subdirectory(flags)
 
-#enable_testing()
-#add_subdirectory(tests)
+enable_testing()
+add_subdirectory(tests)

README.md

@@ -46,6 +46,13 @@ cmake --build build --parallel
 Name from `./models`.
 Image of graph and a dot file are located in `./images`.
 
+## Testing
+
+```bash
+cd build
+ctest --output-on-failure
+```
+
 ### Notes
 
 To get absl:

tests/CMakeLists.txt

@@ -7,4 +7,23 @@ FetchContent_Declare(
     URL https://github.com/google/googletest/archive/refs/tags/v1.17.0.zip
 )
 
-FetchContent_MakeAvailable(googletest)
+FetchContent_MakeAvailable(googletest)
+
+add_executable(parser_test
+    parser_tests.cpp)
+
+target_link_libraries(parser_test
+    PRIVATE
+        onnx_parser
+        GTest::gtest
+)
+
+target_include_directories(parser_test
+    PRIVATE
+        ${CMAKE_CURRENT_SOURCE_DIR}/../frontend/parser/onnx/include
+        ${CMAKE_CURRENT_SOURCE_DIR}/../frontend/ir/include
+        ${CMAKE_BINARY_DIR}
+)
+
+include(GoogleTest)
+gtest_discover_tests(parser_test)

tests/parser_tests.cpp

@@ -0,0 +1,60 @@
+#include <gtest/gtest.h>
+
+#include <filesystem>
+
+#include "parser.hpp"
+#include "types.hpp"
+
+using namespace TensorCompiler;
+
+TEST(SimpleParserTest, AddMulModel) {
+    ONNXParser parser;
+
+    std::string model_path = std::string(PROJECT_ROOT) + "/models/add_mul.onnx";
+
+    std::cout << "Model path: " << model_path << '\n';
+    EXPECT_TRUE(std::filesystem::exists(model_path)) << "File does not exist\n";
+    ASSERT_TRUE(parser.load(model_path)) << "Load failed\n";
+
+    ASSERT_TRUE(parser.load(model_path))
+        << "Couldnot load add_mul.onnx model.\n";
+
+    const auto& graph = parser.ParseGraph();
+
+    EXPECT_EQ(graph.tensors().size(), 5);
+    EXPECT_EQ(graph.nodes().size(), 2);
+
+    std::vector<std::string> expected_tensors = {"input1", "input2", "input3",
+                                                 "add_out", "output"};
+    for (const auto& name : expected_tensors) {
+        const Tensor* t = graph.get_const_tensor(name);
+        EXPECT_NE(t, nullptr) << "Tensor '" << name << "' not found\n";
+    }
+
+    const Node* add_node = nullptr;
+    const Node* mul_node = nullptr;
+    for (const auto& [name, node] : graph.nodes()) {
+        if (node->op_type() == OpType::Add) add_node = node.get();
+        if (node->op_type() == OpType::Mul) mul_node = node.get();
+    }
+
+    ASSERT_NE(add_node, nullptr) << "Add node not found\n";
+    ASSERT_NE(mul_node, nullptr) << "Mul node not found\n";
+
+    EXPECT_EQ(add_node->inputs().size(), 2);
+    EXPECT_EQ(add_node->outputs().size(), 1);
+    EXPECT_EQ(add_node->inputs()[0], "input1");
+    EXPECT_EQ(add_node->inputs()[1], "input2");
+    EXPECT_EQ(add_node->outputs()[0], "add_out");
+
+    EXPECT_EQ(mul_node->inputs().size(), 2);
+    EXPECT_EQ(mul_node->outputs().size(), 1);
+    EXPECT_EQ(mul_node->inputs()[0], "add_out");
+    EXPECT_EQ(mul_node->inputs()[1], "input3");
+    EXPECT_EQ(mul_node->outputs()[0], "output");
+}
+
+int main(int argc, char** argv) {
+    ::testing::InitGoogleTest(&argc, argv);
+    return RUN_ALL_TESTS();
+}