forked from M-Labs/artiq
1
0
Fork 0

lda: add unit tests

This commit is contained in:
Sebastien Bourdeauducq 2015-02-18 11:33:25 -07:00
parent 05824fd3be
commit de4a15c315
1 changed files with 40 additions and 0 deletions

40
artiq/test/lda.py Normal file
View File

@ -0,0 +1,40 @@
import unittest
import os
from artiq.devices.lda.driver import Lda, Ldasim
from artiq.language.units import dB
no_hardware = bool(os.getenv("ARTIQ_NO_HARDWARE"))
class GenericLdaTest:
def test_attenuation(self):
step = self.cont.get_att_step_size().amount
max = self.cont.get_att_max().amount
test_vector = [i*step*dB for i in range(0, int(max*int(1/step)+1))]
for i in test_vector:
with self.subTest(i=i):
self.cont.set_attenuation(i)
self.assertEqual(i, self.cont.get_attenuation())
@unittest.skipIf(no_hardware, "no hardware")
class TestLda(GenericLdaTest, unittest.TestCase):
def setUp(self):
device = os.getenv("ARTIQ_LDA_DEVICE")
serial = os.getenv("ARTIQ_LDA_SERIAL")
args = dict()
if device is not None:
args["product"] = device
if serial is not None:
args["serial"] = serial
self.cont = Lda(**args)
class TestLdaSim(GenericLdaTest, unittest.TestCase):
def setUp(self):
self.cont = Ldasim()
if __name__ == "__main__":
unittest.main()