def test_compare_to_original(self):
original = Signal(np.arange(-20, 20), 20)
frames = _make_frames(200, 150, original)
restored = fr.restore(frames, 20)
for i in range(original.size):
msg = f'Difference at {i}: {original.amps[i]} != {restored[i]}'
self.assertEqual(original.amps[i], restored[i], msg)
def test_append_only_zeros(self):
original = Signal(np.arange(-20, 20), 20)
frames = _make_frames(200, 150, original)
restored = fr.restore(frames, 20)
padding = _padlen(*frames.shape, 150, original)
for idx, pad in enumerate(restored[:-padding]):
msg = f'Non-zero padding at {frames.shape[0] + idx}'
self.assertEqual(0, pad, msg)
def test_empty_frames_2d(self):
restored = fr.restore(np.array([[]]), 8000)
self.assertEqual(restored.size, 0)
def test_empty_frames(self):
restored = fr.restore(np.array([]), 16000)
self.assertEqual(restored.size, 0)
def test_size(self):
frames = _make_frames(25, 10, signal01)
restored = fr.restore(frames, fs1)
padlen = _padlen(*frames.shape, 10, signal01)
self.assertEqual(signal01.size + padlen, restored.size)