Update cocotb examples and tests paths

cocotb 2.0 will move its source to src/ subdirectory in their repository.
themperek · Jan 3, 2025 · 248723c · 248723c
1 parent fc382bb
commit 248723c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/tests/test_cocotb_examples.py b/tests/test_cocotb_examples.py
@@ -2,9 +2,15 @@
 import cocotb
 import pytest
 
+from cocotb_test.compat import cocotb_2x_or_newer
 from cocotb_test.simulator import run
 
-example_dir = os.path.join(os.path.dirname(os.path.dirname(cocotb.__file__)), "examples")
+if cocotb_2x_or_newer:
+    cocotb_repo_root = os.path.dirname(os.path.dirname(os.path.dirname(cocotb.__file__)))
+else:
+    cocotb_repo_root = os.path.dirname(os.path.dirname(cocotb.__file__))
+
+example_dir = os.path.join(cocotb_repo_root, "examples")
 
 if os.path.isdir(example_dir) == False:
     raise IOError("Cocotb example directory not found. Please clone with git and install with `pip -e`")

diff --git a/tests/test_cocotb_tests.py b/tests/test_cocotb_tests.py
@@ -3,9 +3,15 @@
 import pytest
 import sys
 
+from cocotb_test.compat import cocotb_2x_or_newer
 from cocotb_test.simulator import run
 
-tests_dir = os.path.join(os.path.dirname(os.path.dirname(cocotb.__file__)), "tests")
+if cocotb_2x_or_newer:
+    cocotb_repo_root = os.path.dirname(os.path.dirname(os.path.dirname(cocotb.__file__)))
+else:
+    cocotb_repo_root = os.path.dirname(os.path.dirname(cocotb.__file__))
+
+tests_dir = os.path.join(cocotb_repo_root, "tests")
 
 if os.path.isdir(tests_dir) == False:
     raise IOError(