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(start=1.5, stop=3.2, scale=.01))
|
|
|
|
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 matplotlib.pyplot as plt
|
|
y = self.test_run()
|
|
plt.step(np.arange(len(y)), y)
|
|
plt.show()
|