2
0
mirror of https://github.com/m-labs/artiq.git synced 2024-12-25 03:08:27 +08:00

test/thorlabs_tcube: fix test discovery and style

This commit is contained in:
Sebastien Bourdeauducq 2015-03-04 23:53:49 +00:00
parent c2831db253
commit 9fad01d967

View File

@ -5,10 +5,11 @@ import time
from artiq.devices.thorlabs_tcube.driver import Tdc, Tpz, TdcSim, TpzSim
from artiq.language.units import V
no_hardware = bool(os.getenv("ARTIQ_NO_HARDWARE"))
class GenericTdcTest(unittest.TestCase):
class GenericTdcTest:
def test_pot_parameters(self):
test_vector = 1, 2, 3, 4, 5, 6, 7, 8
self.cont.set_pot_parameters(*test_vector)
@ -82,14 +83,13 @@ class GenericTdcTest(unittest.TestCase):
self.assertEqual(test_vector, self.cont.get_button_parameters())
class GenericTpzTest(unittest.TestCase):
class GenericTpzTest:
def test_position_control_mode(self):
test_vector = 1
self.cont.set_position_control_mode(test_vector)
self.assertEqual(test_vector, self.cont.get_position_control_mode())
def test_ouput_volts(self):
for voltage in 5*V, 10*V, 15*V, \
round(self.cont.get_tpz_io_settings()[0].amount)*V:
with self.subTest(voltage=voltage):
@ -135,7 +135,7 @@ class GenericTpzTest(unittest.TestCase):
@unittest.skipIf(no_hardware, "no hardware")
class TestTdc(GenericTdcTest):
class TestTdc(unittest.TestCase, GenericTdcTest):
def setUp(self):
serial = os.getenv("ARTIQ_TDC001_SERIAL")
args = dict()
@ -144,13 +144,13 @@ class TestTdc(GenericTdcTest):
self.cont = Tdc(**args)
class TestTdcSim(GenericTdcTest):
class TestTdcSim(unittest.TestCase, GenericTdcTest):
def setUp(self):
self.cont = TdcSim()
@unittest.skipIf(no_hardware, "no hardware")
class TestTpz(GenericTpzTest):
class TestTpz(unittest.TestCase, GenericTpzTest):
def setUp(self):
serial = os.getenv("ARTIQ_TPZ001_SERIAL")
args = dict()
@ -159,6 +159,6 @@ class TestTpz(GenericTpzTest):
self.cont = Tpz(**args)
class TestTpzSim(GenericTpzTest):
class TestTpzSim(unittest.TestCase, GenericTpzTest):
def setUp(self):
self.cont = TpzSim()