mirror of https://github.com/m-labs/artiq.git
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
from artiq.experiment import *
|
|
from artiq.coredevice.comm_analyzer import (decode_dump, StoppedMessage,
|
|
OutputMessage, InputMessage,
|
|
_extract_log_chars, get_analyzer_dump)
|
|
from artiq.test.hardware_testbench import ExperimentCase
|
|
|
|
|
|
class CreateTTLPulse(EnvExperiment):
|
|
def build(self):
|
|
self.setattr_device("core")
|
|
self.setattr_device("loop_in")
|
|
self.setattr_device("loop_out")
|
|
|
|
@kernel
|
|
def initialize_io(self):
|
|
self.core.reset()
|
|
self.loop_in.input()
|
|
self.loop_out.off()
|
|
|
|
@kernel
|
|
def run(self):
|
|
self.core.break_realtime()
|
|
with parallel:
|
|
self.loop_in.gate_both_mu(1200)
|
|
with sequential:
|
|
delay_mu(100)
|
|
self.loop_out.pulse_mu(1000)
|
|
self.loop_in.count()
|
|
|
|
|
|
class WriteLog(EnvExperiment):
|
|
def build(self):
|
|
self.setattr_device("core")
|
|
|
|
@kernel
|
|
def run(self):
|
|
self.core.reset()
|
|
rtio_log("foo", 32)
|
|
|
|
|
|
class AnalyzerTest(ExperimentCase):
|
|
def test_ttl_pulse(self):
|
|
core_host = self.device_mgr.get_desc("core")["arguments"]["host"]
|
|
|
|
exp = self.create(CreateTTLPulse)
|
|
exp.initialize_io()
|
|
get_analyzer_dump(core_host) # clear analyzer buffer
|
|
exp.run()
|
|
|
|
dump = decode_dump(get_analyzer_dump(core_host))
|
|
self.assertIsInstance(dump.messages[-1], StoppedMessage)
|
|
output_messages = [msg for msg in dump.messages
|
|
if isinstance(msg, OutputMessage)
|
|
and msg.address == 0]
|
|
self.assertEqual(len(output_messages), 2)
|
|
self.assertEqual(
|
|
abs(output_messages[0].timestamp - output_messages[1].timestamp),
|
|
1000)
|
|
input_messages = [msg for msg in dump.messages
|
|
if isinstance(msg, InputMessage)]
|
|
self.assertEqual(len(input_messages), 2)
|
|
self.assertAlmostEqual(
|
|
abs(input_messages[0].timestamp - input_messages[1].timestamp),
|
|
1000, delta=1)
|
|
|
|
def test_rtio_log(self):
|
|
core_host = self.device_mgr.get_desc("core")["arguments"]["host"]
|
|
|
|
exp = self.create(WriteLog)
|
|
get_analyzer_dump(core_host) # clear analyzer buffer
|
|
exp.run()
|
|
|
|
dump = decode_dump(get_analyzer_dump(core_host))
|
|
log = "".join([_extract_log_chars(msg.data)
|
|
for msg in dump.messages
|
|
if isinstance(msg, OutputMessage) and msg.channel == dump.log_channel])
|
|
self.assertEqual(log, "foo\x1E32\x1D")
|