Delete sim_main in __init__ all (#413)

Signed-off-by: yihong0618 <[email protected]> Co-authored-by: Tian Gao <[email protected]>
gaogaotiantian · Mar 19, 2024 · d968411 · d968411
1 parent 49a1dcf
commit d968411
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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)