fix imports

tests/dataset/__init__.py

@@ -1,3 +1,3 @@
 import pytest
 
-pytest.register_assert_rewrite('qcodes.tests.dataset.helper_functions')
+pytest.register_assert_rewrite("tests.dataset.helper_functions")

tests/dataset/dond/test_do0d.py

@@ -14,7 +14,7 @@
     Multi2DSetPointParam2Sizes,
     MultiSetPointParam,
 )
-from qcodes.tests.dataset.conftest import ArrayshapedParam
+from tests.dataset.conftest import ArrayshapedParam
 
 
 @pytest.mark.usefixtures("plot_close", "experiment")

tests/dataset/dond/test_do1d.py

@@ -19,7 +19,7 @@
     MultiSetPointParam,
 )
 from qcodes.parameters import Parameter
-from qcodes.tests.dataset.conftest import ArrayshapedParam
+from tests.dataset.conftest import ArrayshapedParam
 
 
 @pytest.mark.usefixtures("plot_close", "experiment")

tests/dataset/dond/test_doNd.py

@@ -33,8 +33,8 @@
     MultiSetPointParam,
 )
 from qcodes.parameters import ManualParameter, Parameter, ParameterBase
-from qcodes.tests.dataset.conftest import ArrayshapedParam
 from qcodes.validators import Ints
+from tests.dataset.conftest import ArrayshapedParam
 
 
 class TrackingParameter(Parameter):

tests/dataset/measurement/test_measurement_context_manager.py

@@ -25,7 +25,7 @@
 from qcodes.dataset.sqlite.connection import atomic_transaction
 from qcodes.parameters import ManualParameter, Parameter, expand_setpoints_helper
 from qcodes.station import Station
-from qcodes.tests.common import retry_until_does_not_throw
+from tests.common import retry_until_does_not_throw
 
 
 def test_log_messages(caplog: LogCaptureFixture, meas_with_registered_param) -> None:

tests/dataset/test_database_creation_and_upgrading.py

@@ -9,7 +9,7 @@
 
 import qcodes as qc
 import qcodes.dataset.descriptions.versioning.serialization as serial
-import qcodes.tests.dataset
+import tests.dataset
 from qcodes.dataset import (
     ConnectionPlus,
     connect,
@@ -50,10 +50,10 @@
     is_column_in_table,
     one,
 )
-from qcodes.tests.common import error_caused_by, skip_if_no_fixtures
-from qcodes.tests.dataset.conftest import temporarily_copied_DB
+from tests.common import error_caused_by, skip_if_no_fixtures
+from tests.dataset.conftest import temporarily_copied_DB
 
-fixturepath = os.sep.join(qcodes.tests.dataset.__file__.split(os.sep)[:-1])
+fixturepath = os.sep.join(tests.dataset.__file__.split(os.sep)[:-1])
 fixturepath = os.path.join(fixturepath, 'fixtures')
 
 

tests/dataset/test_database_extract_runs.py

@@ -12,7 +12,7 @@
 from numpy.testing import assert_array_equal
 
 import qcodes as qc
-import qcodes.tests.dataset
+import tests.dataset
 from qcodes.dataset import do1d, do2d
 from qcodes.dataset.data_set import (
     DataSet,
@@ -35,7 +35,7 @@
 from qcodes.dataset.sqlite.queries import get_experiments
 from qcodes.instrument_drivers.instrument_mocks import DummyInstrument
 from qcodes.station import Station
-from qcodes.tests.common import error_caused_by, skip_if_no_fixtures
+from tests.common import error_caused_by, skip_if_no_fixtures
 
 
 @contextmanager
@@ -727,7 +727,7 @@ def test_old_versions_not_touched(
 
     _, new_v = get_db_version_and_newest_available_version(source_path)
 
-    fixturepath = os.sep.join(qcodes.tests.dataset.__file__.split(os.sep)[:-1])
+    fixturepath = os.sep.join(tests.dataset.__file__.split(os.sep)[:-1])
     fixturepath = os.path.join(fixturepath,
                                'fixtures', 'db_files', 'version2',
                                'some_runs.db')

tests/dataset/test_dataset_basic.py

@@ -29,10 +29,10 @@
 from qcodes.dataset.sqlite.connection import atomic, path_to_dbfile
 from qcodes.dataset.sqlite.database import _convert_array, get_DB_location
 from qcodes.dataset.sqlite.queries import _rewrite_timestamps, _unicode_categories
-from qcodes.tests.common import error_caused_by
-from qcodes.tests.dataset.helper_functions import verify_data_dict
-from qcodes.tests.dataset.test_links import generate_some_links
 from qcodes.utils.types import complex_types, numpy_complex, numpy_floats, numpy_ints
+from tests.common import error_caused_by
+from tests.dataset.helper_functions import verify_data_dict
+from tests.dataset.test_links import generate_some_links
 
 n_experiments = 0
 

tests/dataset/test_dependencies.py

@@ -12,7 +12,7 @@
 from qcodes.dataset.descriptions.param_spec import ParamSpec, ParamSpecBase
 from qcodes.dataset.descriptions.versioning.converters import new_to_old, old_to_new
 from qcodes.dataset.descriptions.versioning.v0 import InterDependencies
-from qcodes.tests.common import error_caused_by
+from tests.common import error_caused_by
 
 
 def test_wrong_input_raises() -> None:

tests/dataset/test_fix_functions.py

@@ -5,7 +5,7 @@
 
 import qcodes.dataset.descriptions.versioning.serialization as serial
 import qcodes.dataset.descriptions.versioning.v0 as v0
-import qcodes.tests.dataset
+import tests.dataset
 from qcodes.dataset.database_fix_functions import (
     fix_version_4a_run_description_bug,
     fix_wrong_run_descriptions,
@@ -15,10 +15,10 @@
 from qcodes.dataset.descriptions.versioning.converters import old_to_new
 from qcodes.dataset.sqlite.db_upgrades.version import get_user_version
 from qcodes.dataset.sqlite.queries import get_run_description
-from qcodes.tests.common import skip_if_no_fixtures
-from qcodes.tests.dataset.conftest import temporarily_copied_DB
+from tests.common import skip_if_no_fixtures
+from tests.dataset.conftest import temporarily_copied_DB
 
-fixturepath = os.sep.join(qcodes.tests.dataset.__file__.split(os.sep)[:-1])
+fixturepath = os.sep.join(tests.dataset.__file__.split(os.sep)[:-1])
 fixturepath = os.path.join(fixturepath, 'fixtures')
 
 

tests/dataset/test_nested_measurements.py

@@ -10,7 +10,7 @@
 from qcodes.dataset.descriptions.dependencies import InterDependencies_
 from qcodes.dataset.descriptions.param_spec import ParamSpecBase
 from qcodes.dataset.sqlite.connection import atomic_transaction
-from qcodes.tests.common import retry_until_does_not_throw
+from tests.common import retry_until_does_not_throw
 
 VALUE = Union[str, float, list, np.ndarray, bool]
 

tests/dataset/test_sqlite_base.py

@@ -29,8 +29,8 @@
 from qcodes.dataset.sqlite import query_helpers as mut_help
 from qcodes.dataset.sqlite.connection import atomic_transaction, path_to_dbfile
 from qcodes.dataset.sqlite.database import get_DB_location
-from qcodes.tests.common import error_caused_by
 from qcodes.utils import QCoDeSDeprecationWarning
+from tests.common import error_caused_by
 
 from .helper_functions import verify_data_dict
 

tests/dataset/test_sqlite_connection.py

@@ -10,7 +10,7 @@
     make_connection_plus_from,
 )
 from qcodes.dataset.sqlite.database import connect
-from qcodes.tests.common import error_caused_by
+from tests.common import error_caused_by
 
 
 def sqlite_conn_in_transaction(conn: sqlite3.Connection):

tests/dataset/test_subscribing.py

@@ -10,7 +10,7 @@
 from qcodes.dataset.descriptions.dependencies import InterDependencies_
 from qcodes.dataset.descriptions.param_spec import ParamSpecBase
 from qcodes.dataset.sqlite.connection import atomic_transaction
-from qcodes.tests.common import retry_until_does_not_throw
+from tests.common import retry_until_does_not_throw
 
 log = logging.getLogger(__name__)
 
@@ -67,7 +67,7 @@ def _make_working_subscriber_config(tmp_path):
         "subscription":{
             "subscribers":{
                 "test_subscriber":{
-                    "factory": "qcodes.tests.dataset.test_subscribing.MockSubscriber",
+                    "factory": "tests.dataset.test_subscribing.MockSubscriber",
                     "factory_kwargs":{
                         "lg": false
                     },
@@ -96,7 +96,7 @@ def _make_broken_subscriber_config(tmp_path):
         "subscription":{
             "subscribers":{
                 "test_subscriber_wrong":{
-                    "factory": "qcodes.tests.dataset.test_subscribing.MockSubscriber",
+                    "factory": "tests.dataset.test_subscribing.MockSubscriber",
                     "factory_kwargs":{
                         "lg": false
                     },

tests/drivers/test_Agilent_E8257D.py

@@ -1,5 +1,5 @@
 from qcodes.instrument_drivers.agilent import AgilentE8257D
-from qcodes.tests.driver_test_case import DriverTestCase
+from tests.driver_test_case import DriverTestCase
 
 
 class TestAgilentE8257D(DriverTestCase):

tests/drivers/test_lakeshore_335.py

@@ -2,16 +2,15 @@
 import time
 
 from qcodes.instrument import InstrumentBase
-from qcodes.instrument_drivers.Lakeshore import LakeshoreModel335
-
-from .test_lakeshore import (
+from qcodes.instrument_drivers.instrument_mocks import instrument_fixture
+from qcodes.instrument_drivers.instrument_mocks.lakeshore import (
     DictClass,
     MockVisaInstrument,
     command,
-    instrument_fixture,
     query,
     split_args,
 )
+from qcodes.instrument_drivers.Lakeshore import LakeshoreModel335
 
 log = logging.getLogger(__name__)
 

tests/drivers/test_lakeshore_336.py

@@ -2,16 +2,15 @@
 import time
 
 from qcodes.instrument import InstrumentBase
-from qcodes.instrument_drivers.Lakeshore import LakeshoreModel336
-
-from .test_lakeshore import (
+from qcodes.instrument_drivers.instrument_mocks import instrument_fixture
+from qcodes.instrument_drivers.instrument_mocks.lakeshore import (
     DictClass,
     MockVisaInstrument,
     command,
-    instrument_fixture,
     query,
     split_args,
 )
+from qcodes.instrument_drivers.Lakeshore import LakeshoreModel336
 
 log = logging.getLogger(__name__)
 

tests/drivers/test_lakeshore_336_legacy.py

@@ -2,16 +2,15 @@
 import time
 
 from qcodes.instrument import InstrumentBase
-from qcodes.instrument_drivers.Lakeshore.Model_336 import Model_336
-
-from .test_lakeshore import (
+from qcodes.instrument_drivers.instrument_mocks import instrument_fixture
+from qcodes.instrument_drivers.instrument_mocks.lakeshore import (
     DictClass,
     MockVisaInstrument,
     command,
-    instrument_fixture,
     query,
     split_args,
 )
+from qcodes.instrument_drivers.Lakeshore.Model_336 import Model_336
 
 log = logging.getLogger(__name__)
 

tests/drivers/test_tektronix_AWG70000A.py

@@ -12,7 +12,7 @@
 from lxml import etree
 from pytest import LogCaptureFixture
 
-import qcodes.tests.drivers.auxiliary_files as auxfiles
+import tests.drivers.auxiliary_files as auxfiles
 from qcodes.instrument_drivers.tektronix.AWG70000A import AWG70000A
 from qcodes.instrument_drivers.tektronix.AWG70002A import AWG70002A
 

tests/drivers/test_weinchel.py

@@ -1,5 +1,5 @@
 from qcodes.instrument_drivers.weinschel import Weinschel8320
-from qcodes.tests.driver_test_case import DriverTestCase
+from tests.driver_test_case import DriverTestCase
 
 
 class TestWeinschel8320(DriverTestCase):

tests/helpers/test_compare_dictionaries.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 
-from qcodes.tests.common import compare_dictionaries
+from tests.common import compare_dictionaries
 
 
 def test_same() -> None:

tests/test_instrument.py

@@ -344,7 +344,7 @@ class GammyInstrument(Instrument):
     error_msg = (
         "Instrument instr is <class "
         "'qcodes.instrument_drivers.instrument_mocks.DummyInstrument'> but "
-        "<class 'qcodes.tests.test_instrument"
+        "<class 'tests.test_instrument"
         ".test_find_same_name_but_different_class.<locals>"
         ".GammyInstrument'> was requested"
     )

tests/test_station.py

@@ -587,8 +587,9 @@ def test_able_to_load_instrument_with_name_argument_not_being_the_first() -> Non
         """
 instruments:
   name_goes_second:
-    type: qcodes.tests.test_station.InstrumentWithNameAsNotFirstArgument
-        """)
+    type: tests.test_station.InstrumentWithNameAsNotFirstArgument
+        """
+    )
 
     instr = st.load_instrument('name_goes_second', first_arg=42)
     assert isinstance(instr, InstrumentWithNameAsNotFirstArgument)