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..dbf062a8f9d05ef5d661b26e439e9970fd104c8f 100644 --- a/profiler/test/ut/compare_tools/utils/test_file_reader.py +++ b/profiler/test/ut/compare_tools/utils/test_file_reader.py @@ -1,6 +1,7 @@ import unittest from utils.file_reader import FileReader +from utils.constant import Constant class TestFileReader(unittest.TestCase): @@ -15,4 +16,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)