mirror of https://github.com/m-labs/artiq
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# Copyright (C) 2014, 2015 Robert Jordens <jordens@gmail.com>
|
|
|
|
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(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())
|
|
|
|
def test_compare(self):
|
|
scale = 100
|
|
d = list(self.s.get_segment(start=0, stop=4, scale=1/scale))
|
|
d[0]["trigger"] = True
|
|
s = compute_samples.Synthesizer(self.y.shape[0], [d])
|
|
s.select(0)
|
|
y = s.trigger()[0]
|
|
np.testing.assert_almost_equal(y[::scale], self.y[0, :-1])
|
|
|
|
@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()
|