From 8c2f8095dac80688306caefeb6f06c274a1e42a1 Mon Sep 17 00:00:00 2001 From: David de la Iglesia Castro Date: Wed, 26 Jan 2022 16:15:18 +0100 Subject: [PATCH] debug: Remove test_viztracer_internal --- tests/unit/command/test_debug.py | 19 ------------------- 1 file changed, 19 deletions(-) diff --git a/tests/unit/command/test_debug.py b/tests/unit/command/test_debug.py index 6de3c24f2..b97c4499b 100644 --- a/tests/unit/command/test_debug.py +++ b/tests/unit/command/test_debug.py @@ -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"]())