Merge pull request #200 from jenshnielsen/qcodes_37

support qcodes 0.37
QCoDeS · Jan 30, 2023 · 978db22 · 978db22
2 parents 772f832 + 5fd491a
commit 978db22
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
@@ -1,4 +1,8 @@
-from qcodes.instrument_drivers.test import DriverTestCase
+try:
+    from qcodes.tests.driver_test_case import DriverTestCase
+except ImportError:
+    # qcodes < 0.37
+    from qcodes.instrument_drivers.test import DriverTestCase  # type: ignore[import,no-redef]
 import unittest
 
 try: