From 67a6beb1b1bf27c90f7d04610942f9aa01a82cc8 Mon Sep 17 00:00:00 2001
From: huanghui <huanghui44@huawei.com>
Date: Sat, 27 Nov 2021 14:20:59 +0800
Subject: [PATCH] add ut for debug info

---
 tests/ut/cpp/debug/debug_info_test.cc | 89 +++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 tests/ut/cpp/debug/debug_info_test.cc

diff --git a/tests/ut/cpp/debug/debug_info_test.cc b/tests/ut/cpp/debug/debug_info_test.cc
new file mode 100644
index 00000000000..efb5af52fe0
--- /dev/null
+++ b/tests/ut/cpp/debug/debug_info_test.cc
@@ -0,0 +1,89 @@
+/**
+ * Copyright 2021 Huawei Technologies Co., Ltd
+ *
+ * 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 <iostream>
+#include <memory>
+#include <set>
+#include "common/common_test.h"
+#include "utils/info.h"
+#include "utils/trace_base.h"
+#include "ir/anf.h"
+#include "ir/func_graph.h"
+#include "frontend/operator/ops.h"
+#include "base/core_ops.h"
+
+namespace mindspore {
+class TestDebugInfo : public UT::Common {
+ public:
+  TestDebugInfo() {}
+};
+
+// Feature: Debug info
+// Description: Make a location
+// Expectation: make a location with no error
+TEST_F(TestDebugInfo, test_make_location) {
+  LocationPtr loc1 = std::make_shared<Location>("/home/workspace/a.py", 0, 4, 1, 8);
+  std::string s = loc1->ToString(kSourceLineTipDiscard);
+
+  std::string expect_str("In file /home/workspace/a.py(0)\n");
+  ASSERT_TRUE(s == expect_str);
+}
+
+// Feature: Debug info
+// Description: Deduplicate the debug infos which have the same print
+// Expectation: a set of debug infos is deduplicated
+TEST_F(TestDebugInfo, test_location_dedup) {
+  LocationPtr loc1 = std::make_shared<Location>("file1.py", 0, 0, 0, 0);
+  NodeDebugInfoPtr debug_info1 = std::make_shared<NodeDebugInfo>();
+  debug_info1->set_location(loc1);
+
+  LocationPtr loc2 = std::make_shared<Location>("file1.py", 0, 0, 0, 0);
+  NodeDebugInfoPtr debug_info2 = std::make_shared<NodeDebugInfo>();
+  debug_info2->set_location(loc2);
+
+  LocationPtr loc3 = std::make_shared<Location>("file2.py", 0, 0, 0, 0);
+  NodeDebugInfoPtr debug_info3 = std::make_shared<NodeDebugInfo>();
+  debug_info3->set_location(loc3);
+
+  std::set<NodeDebugInfoPtr, DebugInfoCompare> fused_debug_info_set;
+  (void)fused_debug_info_set.emplace(debug_info1);
+  ASSERT_TRUE(fused_debug_info_set.size() == 1);
+
+  (void)fused_debug_info_set.emplace(debug_info2);
+  ASSERT_TRUE(fused_debug_info_set.size() == 1);
+
+  (void)fused_debug_info_set.emplace(debug_info3);
+  ASSERT_TRUE(fused_debug_info_set.size() == 2);
+
+  (void)fused_debug_info_set.emplace(debug_info3);
+  ASSERT_TRUE(fused_debug_info_set.size() == 2);
+}
+
+// Feature: Debug info
+// Description: Test adding a fused debug info
+// Expectation: success
+TEST_F(TestDebugInfo, test_fused_debug_info) {
+  FuncGraphPtr fg = std::make_shared<FuncGraph>();
+  std::vector<AnfNodePtr> inputs;
+  CNodePtr cnode = std::make_shared<CNode>(inputs, fg);
+
+  ASSERT_TRUE(cnode->fused_debug_infos().size() == 0);
+
+  NodeDebugInfoPtr debug_info = std::make_shared<NodeDebugInfo>();
+  cnode->AddFusedDebugInfo(debug_info);
+
+  ASSERT_TRUE(cnode->fused_debug_infos().size() == 1);
+}
+}  // namespace mindspore