class DiracSourceTest(unittest.TestCase):
def setUp(self):
self.s = DiracSource(channels=3)
self.s1 = DiracSource(channels=3, length=1)
def test_params(self):
self.assertEqual(self.s.channels, 3)
self.assertTrue(self.s.endless)
def test_one_sample(self):
chunk = list(self.s1.read())
self.assertEqual(chunk, [[1.0, 1.0, 1.0]])
def test_read(self):
for i, x in enumerate(self.s.read()):
if i == 0:
self.assertEqual(x, [1.0, 1.0, 1.0])
else:
self.assertEqual(x, [0.0, 0.0, 0.0])
if i > 100:
break
def setUp(self):
self.s = DiracSource(channels=3)
self.s1 = DiracSource(channels=3, length=1)
