Move common to gdb

Author: ZzEeKkAa

numba_dpex/tests/debugging/common.py

@@ -5,14 +5,14 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import os
+import pathlib
 import sys
 
 import pytest
 
+import numba_dpex
 from numba_dpex.core import config
 
-from .common import script_path
-
 if config.TESTING_SKIP_NO_DEBUGGING:
     pexpect = pytest.importorskip("pexpect")
 else:
@@ -102,3 +102,57 @@ def set_scheduler_lock(self):
     @staticmethod
     def script_path(script):
         return script_path(script)
+
+
+def script_path(script):
+    package_path = pathlib.Path(numba_dpex.__file__).parent
+    return str(package_path / "examples/debug" / script)
+
+
+def line_number(file_path, text):
+    """Return line number of the text in the file"""
+    with open(file_path, "r") as lines:
+        for line_number, line in enumerate(lines):
+            if text in line:
+                return line_number + 1
+
+    raise RuntimeError(f"Can not find {text} in {file_path}")
+
+
+def breakpoint_by_mark(script, mark, offset=0):
+    """Return breakpoint for the mark in the script
+
+    Example: breakpoint_by_mark("script.py", "Set here") -> "script.py:25"
+    """
+    return f"{script}:{line_number(script_path(script), mark) + offset}"
+
+
+def breakpoint_by_function(script, function):
+    """Return breakpoint for the function in the script"""
+    return breakpoint_by_mark(script, f"def {function}", 1)
+
+
+def setup_breakpoint(
+    app: gdb,
+    breakpoint: str,
+    script=None,
+    expected_location=None,
+    expected_line=None,
+):
+    if not script:
+        script = breakpoint.split(" ")[0].split(":")[0]
+
+    if not expected_location:
+        expected_location = breakpoint.split(" ")[0]
+        if not expected_location.split(":")[-1].isnumeric():
+            expected_location = breakpoint_by_function(
+                script, expected_location.split(":")[-1]
+            )
+
+    app.breakpoint(breakpoint)
+    app.run(script)
+
+    app.child.expect(rf"Thread .* hit Breakpoint .* at {expected_location}")
+
+    if expected_line:
+        app.child.expect(expected_line)

numba_dpex/tests/debugging/gdb.py

@@ -11,7 +11,7 @@
 
 from numba_dpex.tests._helper import skip_no_gdb
 
-from .common import setup_breakpoint
+from .gdb import setup_breakpoint
 
 pytestmark = skip_no_gdb
 

numba_dpex/tests/debugging/test_backtraces.py

@@ -13,7 +13,7 @@
 
 from numba_dpex.tests._helper import skip_no_gdb
 
-from .common import breakpoint_by_function, breakpoint_by_mark, setup_breakpoint
+from .gdb import breakpoint_by_function, breakpoint_by_mark, setup_breakpoint
 
 pytestmark = skip_no_gdb
 

numba_dpex/tests/debugging/test_breakpoints.py

@@ -10,7 +10,7 @@
 
 from numba_dpex.tests._helper import skip_no_gdb
 
-from .common import breakpoint_by_function, breakpoint_by_mark, setup_breakpoint
+from .gdb import breakpoint_by_function, breakpoint_by_mark, setup_breakpoint
 
 pytestmark = skip_no_gdb
 

numba_dpex/tests/debugging/test_common.py

@@ -13,7 +13,7 @@
 
 from numba_dpex.tests._helper import skip_no_gdb
 
-from .common import setup_breakpoint
+from .gdb import setup_breakpoint
 from .test_breakpoints import side_by_side_breakpoint
 
 pytestmark = skip_no_gdb

numba_dpex/tests/debugging/test_info.py

@@ -11,7 +11,7 @@
 
 from numba_dpex.tests._helper import skip_no_gdb
 
-from .common import setup_breakpoint
+from .gdb import setup_breakpoint
 
 pytestmark = skip_no_gdb