diff --git a/profiler/test/ut/compare_tools/utils/test_file_reader.py b/profiler/test/ut/compare_tools/utils/test_file_reader.py index 7e755a69cef58a3d595eeaf3ff6889e4749f5ccc..c8ef4fbeb98624eeff0cf3ffd7a0700920ad2879 100644 --- a/profiler/test/ut/compare_tools/utils/test_file_reader.py +++ b/profiler/test/ut/compare_tools/utils/test_file_reader.py @@ -1,7 +1,7 @@ import unittest from utils.file_reader import FileReader - +from utils.constant import Constant class TestFileReader(unittest.TestCase): @@ -15,4 +15,4 @@ class TestFileReader(unittest.TestCase): def test_check_json_type(self): t = FileReader.check_json_type("resource/event_list.json") - self.assertEqual(t, 1) + self.assertEqual(t, Constant.NPU)