From d968411d9426db7e981e5a7387c7bd0da2136868 Mon Sep 17 00:00:00 2001 From: yihong Date: Tue, 19 Mar 2024 13:35:04 +0800 Subject: [PATCH] Delete sim_main in `__init__` all (#413) Signed-off-by: yihong0618 Co-authored-by: Tian Gao --- src/viztracer/__init__.py | 1 - tests/test_cmdline.py | 4 ++++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/viztracer/__init__.py b/src/viztracer/__init__.py index 853f5e32..c0f68b82 100644 --- a/src/viztracer/__init__.py +++ b/src/viztracer/__init__.py @@ -16,7 +16,6 @@ __all__ = [ "__version__", "main", - "sim_main", "viewer_main", "VizTracer", "ignore_function", diff --git a/tests/test_cmdline.py b/tests/test_cmdline.py index 6118982c..40bd5a64 100644 --- a/tests/test_cmdline.py +++ b/tests/test_cmdline.py @@ -379,6 +379,10 @@ def test_pid_suffix_and_output(self): def test_module(self): self.template(["viztracer", "-m", "numbers"]) + def test_import_star(self): + script = "from viztracer import *" + self.template(["python", "cmdline_test.py"], script=script, expected_output_file=None) + def test_log_gc(self): self.template(["viztracer", "--log_gc", "cmdline_test.py"], script=file_gc)