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)