mirror of https://github.com/m-labs/artiq.git
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
|
import unittest
|
||
|
|
||
|
import numpy as np
|
||
|
|
||
|
from artiq.wavesynth import coefficients, compute_samples
|
||
|
|
||
|
|
||
|
class TestSplineCoef(unittest.TestCase):
|
||
|
def setUp(self):
|
||
|
self.x = np.arange(5.)
|
||
|
self.y = np.sin(2*np.pi*self.x/5) + np.arange(2)[:, None]
|
||
|
self.s = coefficients.SplineSource(self.x, self.y, order=4)
|
||
|
|
||
|
def test_get_segment(self):
|
||
|
return list(self.s.get_segment_data(1.5, 3.2, 1/100.))
|
||
|
|
||
|
def test_synth(self):
|
||
|
d = self.test_get_segment()
|
||
|
d[0]["trigger"] = True
|
||
|
return compute_samples.Synthesizer(self.y.shape[0], [d, d + d])
|
||
|
|
||
|
def drive(self, s):
|
||
|
y = []
|
||
|
for f in 0, 1, None, 0:
|
||
|
if f is not None:
|
||
|
s.select(f)
|
||
|
y += s.trigger()[0]
|
||
|
return y
|
||
|
|
||
|
def test_run(self):
|
||
|
return self.drive(self.test_synth())
|
||
|
|
||
|
@unittest.skip("manual/visual test")
|
||
|
def test_plot(self):
|
||
|
import cairoplot
|
||
|
y = self.test_run()
|
||
|
x = list(range(len(y)))
|
||
|
cairoplot.scatter_plot("plot.png", [x, y])
|