From 975346544f0211ebe2a44dc8882f2167cfd1d689 Mon Sep 17 00:00:00 2001
From: shawn_zhu1 <14535601+shawnzhu1@user.noreply.gitee.com>
Date: Thu, 27 Jun 2024 10:15:35 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90bugfix=E3=80=91=E4=BF=AE=E5=A4=8D?=
 =?UTF-8?q?=E5=88=9B=E5=BB=BA=E6=95=B0=E6=8D=AE=E6=96=87=E4=BB=B6=E5=A4=B9?=
 =?UTF-8?q?=E7=9A=84=E6=9D=83=E9=99=90=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py  | 2 +-
 debug/accuracy_tools/ptdbg_ascend/test/run_ut.py          | 8 +++++---
 debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor.py | 3 +--
 .../ptdbg_ascend/test/ut/test_advisor_result.py           | 2 +-
 4 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py
index e257a4db20..c021f4dcf2 100644
--- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py
+++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py
@@ -621,7 +621,7 @@ def save_numpy_data(file_path, data):
     save_numpy_data
     """
     if not os.path.exists(os.path.dirname(file_path)):
-        os.makedirs(os.path.dirname(file_path))
+        os.makedirs(os.path.dirname(file_path), mode=FileCheckConst.DATA_DIR_AUTHORITY)
     np.save(file_path, data)
 
 
diff --git a/debug/accuracy_tools/ptdbg_ascend/test/run_ut.py b/debug/accuracy_tools/ptdbg_ascend/test/run_ut.py
index 5972e73d27..3ca687e4a3 100644
--- a/debug/accuracy_tools/ptdbg_ascend/test/run_ut.py
+++ b/debug/accuracy_tools/ptdbg_ascend/test/run_ut.py
@@ -3,6 +3,7 @@ import shutil
 import subprocess
 import sys
 
+
 def run_ut():
     cur_dir = os.path.realpath(os.path.dirname(__file__))
     top_dir = os.path.realpath(os.path.dirname(cur_dir))
@@ -13,11 +14,11 @@ def run_ut():
     if os.path.exists(report_dir):
         shutil.rmtree(report_dir)
 
-    os.makedirs(report_dir)
+    os.makedirs(report_dir, mode=0o700)
 
     cmd = ["python3", "-m", "pytest", ut_path, "--junitxml=" + report_dir + "/final.xml",
            "--cov=" + src_dir, "--cov-branch", "--cov-report=xml:" + report_dir + "/coverage.xml"]
-    
+
     result_ut = subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
 
     while result_ut.poll() is None:
@@ -34,7 +35,8 @@ def run_ut():
 
     return ut_flag
 
-if __name__=="__main__":
+
+if __name__ == "__main__":
     if run_ut():
         sys.exit(0)
     else:
diff --git a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor.py b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor.py
index 5f96cfe2c2..c33f830b90 100644
--- a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor.py
+++ b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor.py
@@ -12,13 +12,12 @@ import pandas
 
 class TestAdvisor(unittest.TestCase):
     def setUp(self) -> None:
-        os.makedirs("test_result/output", exist_ok=True)
+        os.makedirs("test_result/output", mode=0o700, exist_ok=True)
         self.output_path = os.path.abspath("test_result/output")
 
     def tearDown(self) -> None:
         shutil.rmtree("test_result/", ignore_errors=True)
 
-
     def test_analysis_when_csv_is_valid(self):
         input_data = pandas.read_csv("resources/compare/compare_result_20230703104808.csv")
         advisor = Advisor(input_data, self.output_path)
diff --git a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor_result.py b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor_result.py
index c2ff666a70..80d3c3c450 100644
--- a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor_result.py
+++ b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_advisor_result.py
@@ -11,7 +11,7 @@ import pandas
 
 class TestAdvisor(unittest.TestCase):
     def setUp(self) -> None:
-        os.makedirs("test_result/output", exist_ok=True)
+        os.makedirs("test_result/output", mode=0o700, exist_ok=True)
         self.output_path = os.path.abspath("test_result/output")
         self.has_error = False
 
-- 
Gitee