Skip to content

Commit

Permalink
Clean up warnings
Browse files Browse the repository at this point in the history
  • Loading branch information
jpsecher committed Jan 9, 2024
1 parent 1e72709 commit b3b461d
Show file tree
Hide file tree
Showing 13 changed files with 34 additions and 33 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ def test_various_operations_have_common_functions(qdac): # noqa
qdac.ch02.sine_wave(),
qdac.ch03.triangle_wave(),
qdac.ch04.dc_sweep(start_V=-1, stop_V=1, points=11),
qdac.ch05.dc_list(voltages=(-1,0,1)),
qdac.ch05.dc_list(voltages=(-1, 0, 1)),
qdac.ch05.measurement()
]
# -----------------------------------------------------------------------
Expand Down
10 changes: 5 additions & 5 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_array.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import pytest
from unittest.mock import MagicMock, call
from unittest.mock import call
from qcodes_contrib_drivers.drivers.QDevil.QDAC2 import QDac2
from qcodes_contrib_drivers.drivers.QDevil.QDAC2_Array import QDac2_Array
from .sim_qdac2_fixtures import qdac, qdac2 # noqa
Expand Down Expand Up @@ -96,7 +96,7 @@ def test_fails_on_non_unique_contact_names(qdac, qdac2): # noqa
listener: {'sensorA': 2, 'plungerB': 3}
})
# -----------------------------------------------------------------------
assert f'Contact name sensorA used multiple times' in repr(error)
assert 'Contact name sensorA used multiple times' in repr(error)


def test_internal_connect_to_trigger_out(qdac, qdac2): # noqa
Expand Down Expand Up @@ -176,7 +176,7 @@ def test_set_virtual_voltages_goes_to_correct_qdac(qdac, qdac2): # noqa


def test_sync_steady_state(qdac, qdac2, mocker): # noqa
sleep_s = mocker.patch('qcodes_contrib_drivers.drivers.QDevil.QDAC2_Array.sleep_s') # Don't sleep
sleep_s = mocker.patch('qcodes_contrib_drivers.drivers.QDevil.QDAC2_Array.sleep_s') # Don't sleep
qdacs, controller, listener = two_qdacs(qdac, qdac2)
contacts = {controller: {'A': 2, 'B': 1}, listener: {'C': 3}}
arrangement = qdacs.arrange(contacts)
Expand Down Expand Up @@ -208,7 +208,7 @@ def test_sync_steady_state(qdac, qdac2, mocker): # noqa


def test_sync_leakage(qdac, qdac2, mocker): # noqa
sleep_s = mocker.patch('qcodes_contrib_drivers.drivers.QDevil.QDAC2_Array.sleep_s') # Don't sleep
mocker.patch('qcodes_contrib_drivers.drivers.QDevil.QDAC2_Array.sleep_s') # Don't sleep
qdacs, controller, listener = two_qdacs(qdac, qdac2)
contacts = {controller: {'A': 3}, listener: {'B': 1, 'C': 2}}
arrangement = qdacs.arrange(contacts)
Expand Down Expand Up @@ -298,7 +298,7 @@ def test_frees_internal_triggers_on_exit(qdac, qdac2): # noqa
qdacs, controller, listener = two_qdacs(qdac, qdac2)
contacts = {controller: {'A': 2, 'B': 1}, listener: {'C': 3}}
# -----------------------------------------------------------------------
with qdacs.arrange(contacts, internal_triggers=['starter']) as arrangement:
with qdacs.arrange(contacts, internal_triggers=['starter']):
pass
# -----------------------------------------------------------------------
assert qdac.n_triggers() == len(qdac._internal_triggers)
8 changes: 4 additions & 4 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_awg.py
Original file line number Diff line number Diff line change
Expand Up @@ -292,7 +292,7 @@ def test_awg_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour5:awg:trig:sour int{trigger.value}',
f'sour5:awg:init:cont on',
'sour5:awg:init:cont on',
]


Expand All @@ -306,7 +306,7 @@ def test_awg_trigger_once_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour5:awg:trig:sour int{trigger.value}',
f'sour5:awg:init:cont off',
'sour5:awg:init:cont off',
]


Expand All @@ -320,7 +320,7 @@ def test_awg_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour5:awg:trig:sour ext{trigger}',
f'sour5:awg:init:cont on',
'sour5:awg:init:cont on',
]


Expand All @@ -334,5 +334,5 @@ def test_awg_trigger_once_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour5:awg:trig:sour ext{trigger}',
f'sour5:awg:init:cont off',
'sour5:awg:init:cont off',
]
10 changes: 5 additions & 5 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_current.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def test_measurement_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sens2:trig:sour int{trigger.value}',
f'sens2:init:cont on',
'sens2:init:cont on',
]


Expand All @@ -80,7 +80,7 @@ def test_measurement_trigger_once_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sens2:trig:sour int{trigger.value}',
f'sens2:init:cont off',
'sens2:init:cont off',
]


Expand All @@ -106,7 +106,7 @@ def test_measurement_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sens2:trig:sour ext{trigger}',
f'sens2:init:cont on',
'sens2:init:cont on',
]


Expand All @@ -119,7 +119,7 @@ def test_measurement_trigger_once_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sens2:trig:sour ext{trigger}',
f'sens2:init:cont off',
'sens2:init:cont off',
]


Expand Down Expand Up @@ -352,7 +352,7 @@ def test_current_trigger_on_internal(qdac): # noqa

def test_current_cleanup_triggering_on_exit(qdac): # noqa
# -----------------------------------------------------------------------
with qdac.ch02.measurement() as measurement:
with qdac.ch02.measurement():
qdac.start_recording_scpi()
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
Expand Down
1 change: 1 addition & 0 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_init.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from qcodes.instrument.base import Instrument
from .sim_qdac2_fixtures import visalib


@pytest.mark.parametrize('version,components', [
('3-0.9.6', ['3', '0.9.6']),
('10.2-1.14', ['10.2', '1.14'])
Expand Down
4 changes: 2 additions & 2 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_leakage.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@ def test_diff_matrix():
assert np.allclose(diff, expected)


def test_arrangement_steady_state(qdac, mocker):
sleep_s = mocker.patch('qcodes_contrib_drivers.drivers.QDevil.QDAC2.sleep_s') # Don't sleep
def test_arrangement_steady_state(qdac, mocker): # noqa
sleep_s = mocker.patch('qcodes_contrib_drivers.drivers.QDevil.QDAC2.sleep_s') # Don't sleep
gates = {'sensor1': 1, 'plunger2': 2, 'plunger3': 3}
arrangement = qdac.arrange(gates)
qdac.start_recording_scpi()
Expand Down
4 changes: 2 additions & 2 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_sine.py
Original file line number Diff line number Diff line change
Expand Up @@ -233,7 +233,7 @@ def test_sine_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:sine:trig:sour int{trigger.value}',
f'sour1:sine:init:cont on',
'sour1:sine:init:cont on',
]


Expand All @@ -246,5 +246,5 @@ def test_sine_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:sine:trig:sour ext{trigger}',
f'sour1:sine:init:cont on',
'sour1:sine:init:cont on',
]
Original file line number Diff line number Diff line change
Expand Up @@ -30,31 +30,31 @@ def test_voltage_range_low_min(qdac): # noqa
voltage = qdac.ch02.output_low_range_minimum_V()
# -----------------------------------------------------------------------
assert -10 < voltage < 10
assert qdac.get_recorded_scpi_commands() == [f'sour2:rang:low:min?']
assert qdac.get_recorded_scpi_commands() == ['sour2:rang:low:min?']


def test_voltage_range_low_max(qdac): # noqa
# -----------------------------------------------------------------------
voltage = qdac.ch02.output_low_range_maximum_V()
# -----------------------------------------------------------------------
assert -10 < voltage < 10
assert qdac.get_recorded_scpi_commands() == [f'sour2:rang:low:max?']
assert qdac.get_recorded_scpi_commands() == ['sour2:rang:low:max?']


def test_voltage_range_high_min(qdac): # noqa
# -----------------------------------------------------------------------
voltage = qdac.ch02.output_high_range_minimum_V()
# -----------------------------------------------------------------------
assert -10 < voltage < 10
assert qdac.get_recorded_scpi_commands() == [f'sour2:rang:high:min?']
assert qdac.get_recorded_scpi_commands() == ['sour2:rang:high:min?']


def test_voltage_range_high_max(qdac): # noqa
# -----------------------------------------------------------------------
voltage = qdac.ch02.output_high_range_maximum_V()
# -----------------------------------------------------------------------
assert -10 < voltage < 10
assert qdac.get_recorded_scpi_commands() == [f'sour2:rang:high:max?']
assert qdac.get_recorded_scpi_commands() == ['sour2:rang:high:max?']


@pytest.mark.parametrize('selector', ['dc', 'med', 'high'])
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -230,7 +230,7 @@ def test_list_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:dc:trig:sour int{trigger.value}',
f'sour1:dc:init:cont on',
'sour1:dc:init:cont on',
]


Expand All @@ -243,7 +243,7 @@ def test_list_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:dc:trig:sour ext{trigger}',
f'sour1:dc:init:cont on',
'sour1:dc:init:cont on',
]


Expand Down Expand Up @@ -276,7 +276,7 @@ def test_list_get_voltages(qdac): # noqa

def test_list_cleanup_triggering_on_exit(qdac): # noqa
# -----------------------------------------------------------------------
with qdac.ch01.dc_list(voltages=range(1, 5)) as dc_list:
with qdac.ch01.dc_list(voltages=range(1, 5)):
qdac.start_recording_scpi()
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -235,7 +235,7 @@ def test_sweep_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:dc:trig:sour int{trigger.value}',
f'sour1:dc:init:cont on',
'sour1:dc:init:cont on',
]


Expand All @@ -248,7 +248,7 @@ def test_sweep_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:dc:trig:sour ext{trigger}',
f'sour1:dc:init:cont on',
'sour1:dc:init:cont on',
]


Expand Down
4 changes: 2 additions & 2 deletions qcodes_contrib_drivers/tests/QDevil/test_sim_qdac2_square.py
Original file line number Diff line number Diff line change
Expand Up @@ -286,7 +286,7 @@ def test_square_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:squ:trig:sour int{trigger.value}',
f'sour1:squ:init:cont on',
'sour1:squ:init:cont on',
]


Expand All @@ -299,7 +299,7 @@ def test_square_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:squ:trig:sour ext{trigger}',
f'sour1:squ:init:cont on',
'sour1:squ:init:cont on',
]


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -260,7 +260,7 @@ def test_triangle_trigger_on_internal(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:tri:trig:sour int{trigger.value}',
f'sour1:tri:init:cont on',
'sour1:tri:init:cont on',
]


Expand All @@ -273,5 +273,5 @@ def test_triangle_trigger_on_external(qdac): # noqa
# -----------------------------------------------------------------------
assert qdac.get_recorded_scpi_commands() == [
f'sour1:tri:trig:sour ext{trigger}',
f'sour1:tri:init:cont on',
'sour1:tri:init:cont on',
]
Original file line number Diff line number Diff line change
Expand Up @@ -323,7 +323,7 @@ def test_arrangement_sweep_outer_trigger(qdac): # noqa
outer_trigger_channel=1)
qdac.start_recording_scpi()
# -----------------------------------------------------------------------
sweep = arrangement.virtual_sweep2d(
arrangement.virtual_sweep2d(
inner_contact='plunger1',
inner_voltages=np.linspace(-1, 1, 5),
outer_contact='plunger2',
Expand Down

0 comments on commit b3b461d

Please sign in to comment.