Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

debug: Remove test_viztracer_internal

Browse Source
This commit is contained in:
David de la Iglesia Castro
2022-01-27 09:36:06 +05:45
committed by Saugat Pachhai
parent 5e80fcce4f
commit 8c2f8095da
1 changed files with 0 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/command/test_debug.py
-19
View File
@@ -17,22 +17,3 @@ def test_viztracer(tmp_dir, dvc, mocker):
assert main(["status", "--viztracer-depth", "2"]) == 0
args = viztracer_profile.call_args[1]
assert args["max_stack_depth"] == 2
def test_viztracer_internal(tmp_dir, dvc, mocker):
import sys
viztracer = mocker.MagicMock()
sys.modules["viztracer"] = viztracer
from dvc import _debug
profile = mocker.spy(_debug, "viztracer_profile")
tracer = viztracer.VizTracer.return_value
assert main(["status", "--viztracer"]) == 0
args = viztracer.VizTracer.call_args[1]
assert args["max_stack_depth"] == -1
tracer.start.assert_called_once()
tracer.stop.assert_called_once()
tracer.save.assert_called_once_with(profile.call_args[1]["path"]())