From 49eb5b73b2b0a36a928d6b45835a03b3a49cecc4 Mon Sep 17 00:00:00 2001
From: "Jens H. Nielsen" <Jens.Nielsen@microsoft.com>
Date: Mon, 30 Jan 2023 08:57:53 +0100
Subject: [PATCH] support qcodes 0.37

---
 qcodes_contrib_drivers/tests/test_Keysight_M3201A.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
index e7170938a..2c4f372ec 100644
--- 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
 import unittest
 
 try: