Fix tests for newer line_profiler version

Nodd · Sep 10, 2023 · 1d649a6 · 1d649a6
1 parent 0f5e727
commit 1d649a6
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/tests/test_mainwindow.py b/tests/test_mainwindow.py
@@ -30,8 +30,8 @@ def profiled_function():
         lprof_path = Path(win.config.stats)
         assert lprof_path.is_file()
         assert (
-            win.dockOutputWidget.outputWidget.toPlainText()
-            == f"Wrote profile results to {lprof_path}\n"
+            win.dockOutputWidget.outputWidget.toPlainText().split("\n")[0]
+            == f"Wrote profile results to {lprof_path}"
         )
 
         tree = win.resultsTreeWidget
@@ -87,8 +87,8 @@ def profiled_function2():
         lprof_path = Path(win.config.stats)
         assert lprof_path.is_file()
         assert (
-            win.dockOutputWidget.outputWidget.toPlainText()
-            == f"Wrote profile results to {lprof_path}\n"
+            win.dockOutputWidget.outputWidget.toPlainText().split("\n")[0]
+            == f"Wrote profile results to {lprof_path}"
         )
 
         tree = win.resultsTreeWidget
@@ -106,8 +106,8 @@ def not_profiled_function():
         lprof_path = Path(win.config.stats)
         assert lprof_path.is_file()
         assert (
-            win.dockOutputWidget.outputWidget.toPlainText()
-            == f"Wrote profile results to {lprof_path}\n"
+            win.dockOutputWidget.outputWidget.toPlainText().split("\n")[0]
+            == f"Wrote profile results to {lprof_path}"
         )
 
         tree = win.resultsTreeWidget
@@ -127,8 +127,8 @@ def not_profiled_function():
         lprof_path = Path(win.config.stats)
         assert lprof_path.is_file()
         assert (
-            win.dockOutputWidget.outputWidget.toPlainText()
-            == f"Wrote profile results to {lprof_path}\n"
+            win.dockOutputWidget.outputWidget.toPlainText().split("\n")[0]
+            == f"Wrote profile results to {lprof_path}"
         )
 
         tree = win.resultsTreeWidget