diff --git a/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_cli.py b/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_cli.py index 2672feebf40891e40b2f93a903a4456a981c7bff..fec8f842b08972f77ac3ae35c65b5ad71e834cac 100644 --- a/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_cli.py +++ b/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_cli.py @@ -1,9 +1,15 @@ import unittest from unittest.mock import patch -from msprobe.pytorch.parse_tool.cli import parse +from msprobe.pytorch.parse_tool.cli import parse, _run_interactive_cli class TestCli(unittest.TestCase): + @patch('msprobe.pytorch.parse_tool.cli.InteractiveCli.cmdloop', return_value=None) + def test_run_interactive_cli(self, mock_cmdloop): + _run_interactive_cli() + + mock_cmdloop.assert_called_once() + @patch('msprobe.pytorch.parse_tool.cli._run_interactive_cli') def test_parse(self, mock_run_interactive_cli): parse() diff --git a/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_file_desc.py b/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_file_desc.py new file mode 100644 index 0000000000000000000000000000000000000000..7e272776e9b913f8997dbc8be9605cde49c3eaf6 --- /dev/null +++ b/debug/accuracy_tools/msprobe/test/pytorch_ut/parse_tool/test_file_desc.py @@ -0,0 +1,9 @@ +import unittest + +from msprobe.pytorch.parse_tool.lib.file_desc import FileDesc, NpuDumpFileDesc, DumpDecodeFileDesc + +class test(unittest.TestCase): + def test_initialization(self): + self.FileDesc = FileDesc('name', 'path', timestamp=1) + self.NpuDumpFileDesc = NpuDumpFileDesc('name', 'path', 1, 'op_name', 'op_type', 1) + self.DumpDecodeFileDesc = DumpDecodeFileDesc('name', 'path', 1, 'op_name', 'op_type', 1, 'anchor_type', 1)