fix test

spyder-ide · Jan 27, 2022 · d817a60 · d817a60
1 parent d117c4c
commit d817a60
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/spyder_line_profiler/tests/test_lineprofiler.py b/spyder_line_profiler/tests/test_lineprofiler.py
@@ -17,8 +17,7 @@
 MAIN_APP = qapplication() 
 
 # Local imports
-from spyder_line_profiler.widgets.lineprofiler import LineProfilerWidget
-
+from spyder_line_profiler.spyder.widgets import SpyderLineProfilerWidget
 try:
     from unittest.mock import Mock
 except ImportError:
@@ -44,10 +43,10 @@ def test_profile_and_display_results(qtbot, tmpdir, monkeypatch):
         f.write(TEST_SCRIPT)
 
     MockQMessageBox = Mock()
-    monkeypatch.setattr('spyder_line_profiler.widgets.lineprofiler.QMessageBox',
-                        MockQMessageBox)
+    # monkeypatch.setattr('spyder_line_profiler.widgets.lineprofiler.QMessageBox',
+    #                     MockQMessageBox)
 
-    widget = LineProfilerWidget(None)
+    widget = SpyderLineProfilerWidget(None)
     qtbot.addWidget(widget)
     with qtbot.waitSignal(widget.sig_finished, timeout=10000, raising=True):
         widget.analyze(testfilename)