kopia lustrzana https://github.com/dnet/pySSTV
80 wiersze
2.7 KiB
Python
80 wiersze
2.7 KiB
Python
import unittest
|
|
from itertools import islice, izip
|
|
import pickle
|
|
import mock
|
|
from mock import MagicMock
|
|
from StringIO import StringIO
|
|
import hashlib
|
|
|
|
from pysstv import sstv
|
|
from pysstv.sstv import SSTV
|
|
from common import get_asset_filename
|
|
|
|
|
|
class TestSSTV(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.s = SSTV(False, 48000, 16)
|
|
self.s.VIS_CODE = 0x00
|
|
self.s.SYNC = 7
|
|
|
|
def test_horizontal_sync(self):
|
|
horizontal_sync = self.s.horizontal_sync()
|
|
expected = (1200, self.s.SYNC)
|
|
actual = horizontal_sync.next()
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_gen_freq_bits(self):
|
|
gen_freq_bits = self.s.gen_freq_bits()
|
|
expected = [(1900, 300),
|
|
(1200, 10),
|
|
(1900, 300),
|
|
(1200, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1300, 30),
|
|
(1200, 30)]
|
|
actual = list(islice(gen_freq_bits, 0, 1000))
|
|
self.assertEqual(expected, actual)
|
|
|
|
# FIXME: Instead of using a test fixture, 'expected' should be synthesized?
|
|
def test_gen_values(self):
|
|
gen_values = self.s.gen_values()
|
|
expected = pickle.load(open(get_asset_filename("SSTV_gen_values.p")))
|
|
for e, g in izip(expected, gen_values):
|
|
self.assertAlmostEqual(e, g, delta=0.000000001)
|
|
|
|
def test_gen_samples(self):
|
|
gen_values = self.s.gen_samples()
|
|
# gen_samples uses random to avoid quantization noise
|
|
# by using additive noise, so there's always a chance
|
|
# of running the code two consecutive times on the same machine
|
|
# and having different results.
|
|
# https://en.wikipedia.org/wiki/Quantization_%28signal_processing%29
|
|
sstv.random = MagicMock(return_value=0.4) # xkcd:221
|
|
expected = pickle.load(open(get_asset_filename("SSTV_gen_samples.p")))
|
|
actual = list(islice(gen_values, 0, 1000))
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_write_wav(self):
|
|
self.maxDiff = None
|
|
sio = StringIO()
|
|
sio.close = MagicMock() # ignore close() so we can .getvalue()
|
|
mock_open = MagicMock(return_value=sio)
|
|
with mock.patch('__builtin__.open', mock_open):
|
|
self.s.write_wav('unittest.wav')
|
|
expected = 'bf61c82e96aed1370d5c1753d87729db'
|
|
data = sio.getvalue()
|
|
actual = hashlib.md5(data).hexdigest()
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_init(self):
|
|
self.assertEqual(self.s.image, False)
|
|
self.assertEqual(self.s.samples_per_sec, 48000)
|
|
self.assertEqual(self.s.bits, 16)
|