M-Labs/artiq
2
0
Fork 3
mirror of https://github.com/m-labs/artiq.git synced 2025-02-03 06:10:18 +08:00
Code Issues Projects Releases Wiki Activity

Update test

Browse Source
This commit is contained in:
newell 2024-12-23 16:19:42 -08:00
parent fd639125d1
commit 9de383bdc2
1 changed files with 112 additions and 121 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
artiq/test/coredevice/test_urukul.py
View File

@ -1,12 +1,14 @@
from artiq.coredevice import urukul
from artiq.coredevice.urukul import STA_PROTO_REV_9, urukul_sta_rf_sw
from artiq.experiment import *
from artiq.test.hardware_testbench import ExperimentCase
CPLD = "urukul_cpld"
class UrukulExp(EnvExperiment):
def build(self, runner):
self.setattr_device("core")
self.dev = self.get_device("urukul_cpld")
self.dev = self.get_device(CPLD)
self.runner = runner
def run(self):
@ -73,7 +75,6 @@ class UrukulExp(EnvExperiment):
@kernel
def att_enables(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
self.dev.cfg_att_en(0, False)
@ -83,7 +84,6 @@ class UrukulExp(EnvExperiment):
@kernel
def att_enable_speed(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
n = 10
@ -93,7 +93,6 @@ class UrukulExp(EnvExperiment):
self.set_dataset(
"dt", self.core.mu_to_seconds(self.core.get_rtio_counter_mu() - t0) / n
)
self.set_dataset("dt", None)
@kernel
def att(self):
@ -165,7 +164,6 @@ class UrukulExp(EnvExperiment):
@kernel
def osk(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
self.dev.cfg_osk(0, False)
@ -175,7 +173,6 @@ class UrukulExp(EnvExperiment):
@kernel
def osk_speed(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
n = 10
@ -185,11 +182,9 @@ class UrukulExp(EnvExperiment):
self.set_dataset(
"dt", self.core.mu_to_seconds(self.core.get_rtio_counter_mu() - t0) / n
)
self.set_dataset("dt", None)
@kernel
def drctl(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
self.dev.cfg_drctl(0, False)
@ -199,7 +194,6 @@ class UrukulExp(EnvExperiment):
@kernel
def drctl_speed(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
n = 10
@ -209,11 +203,9 @@ class UrukulExp(EnvExperiment):
self.set_dataset(
"dt", self.core.mu_to_seconds(self.core.get_rtio_counter_mu() - t0) / n
)
self.set_dataset("dt", None)
@kernel
def drhold(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
self.dev.cfg_drhold(0, False)
@ -223,7 +215,6 @@ class UrukulExp(EnvExperiment):
@kernel
def drhold_speed(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
n = 10
@ -233,11 +224,9 @@ class UrukulExp(EnvExperiment):
self.set_dataset(
"dt", self.core.mu_to_seconds(self.core.get_rtio_counter_mu() - t0) / n
)
self.set_dataset("dt", None)
@kernel
def mask_nu(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
self.dev.cfg_mask_nu(0, False)
@ -247,7 +236,6 @@ class UrukulExp(EnvExperiment):
@kernel
def mask_nu_speed(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
n = 10
@ -257,7 +245,6 @@ class UrukulExp(EnvExperiment):
self.set_dataset(
"dt", self.core.mu_to_seconds(self.core.get_rtio_counter_mu() - t0) / n
)
self.set_dataset("dt", None)
# Note, cfg_io_update is tested in test_ad9910.py
@kernel
@ -281,7 +268,6 @@ class UrukulExp(EnvExperiment):
@kernel
def cfg_profile(self):
if self.dev.proto_rev == urukul.STA_PROTO_REV_9:
self.core.break_realtime()
self.dev.init()
self.dev.set_profile(0, 7)
@ -320,17 +306,18 @@ class UrukulTest(ExperimentCase):
def test_switches_readback(self):
self.execute(UrukulExp, "switches_readback")
sw_get = urukul.urukul_sta_rf_sw(self.dataset_mgr.get("sta_get"))
sw_get = urukul_sta_rf_sw(self.dataset_mgr.get("sta_get"))
sw_set = self.dataset_mgr.get("sw_set")
self.assertEqual(sw_get, sw_set)
def test_att_enables(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "att_enables")
def test_att_enable_speed(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "att_enable_speed")
dt = self.dataset_mgr.get("dt")
if dt:
print(dt)
self.assertLess(dt, 5 * us)
@ -362,42 +349,46 @@ class UrukulTest(ExperimentCase):
self.assertLess(dt, 5 * us)
def test_osk(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "osk")
def test_osk_speed(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "osk_speed")
dt = self.dataset_mgr.get("dt")
if dt:
print(dt)
self.assertLess(dt, 5 * us)
def test_drctl(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "drctl")
def test_drctl_speed(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "drctl_speed")
dt = self.dataset_mgr.get("dt")
if dt:
print(dt)
self.assertLess(dt, 5 * us)
def test_drhold(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "drhold")
def test_drhold_speed(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "drhold_speed")
dt = self.dataset_mgr.get("dt")
if dt:
print(dt)
self.assertLess(dt, 5 * us)
def test_mask_nu(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "mask_nu")
def test_mask_nu_speed(self):
if self.device_mgr.get(CPLD).proto_rev == STA_PROTO_REV_9:
self.execute(UrukulExp, "mask_nu_speed")
dt = self.dataset_mgr.get("dt")
if dt:
print(dt)
self.assertLess(dt, 5 * us)