dnet/pySSTV
1
0
Fork 0
mirror of https://github.com/dnet/pySSTV.git synced 2026-01-11 11:10:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed relative filenames in tests

Browse source
This commit is contained in:
András Veres-Szentkirályi 2013-06-24 13:41:35 +02:00
parent 391377551c
commit 1b592a4ad5
3 changed files with 15 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pysstv/tests/common.py Normal file
View file

@ -0,0 +1,4 @@
from os import path
def get_asset_filename(filename):
return path.join(path.dirname(__file__), 'assets', filename)

11
pysstv/tests/test_color.py
View file

@ -4,7 +4,8 @@ import pickle
from PIL import Image
import color
from pysstv import color
from common import get_asset_filename
class TestMartinM1(unittest.TestCase):
@ -12,16 +13,16 @@ class TestMartinM1(unittest.TestCase):
def setUp(self):
self.image = Image.new('RGB', (320, 256))
self.s = color.MartinM1(self.image, 48000, 16)
lena = Image.open('tests/assets/320x256.png')
lena = Image.open(get_asset_filename('320x256.png'))
self.lena = color.MartinM1(lena, 48000, 16)
def test_gen_freq_bits(self):
expected = pickle.load(open("tests/assets/MartinM1_freq_bits.p"))
expected = pickle.load(open(get_asset_filename("MartinM1_freq_bits.p")))
actual = list(islice(self.s.gen_freq_bits(), 0, 1000))
self.assertEqual(expected, actual)
def test_gen_freq_bits_lena(self):
expected = pickle.load(open("tests/assets/MartinM1_freq_bits_lena.p"))
expected = pickle.load(open(get_asset_filename("MartinM1_freq_bits_lena.p")))
actual = list(islice(self.lena.gen_freq_bits(), 0, 1000))
self.assertEqual(expected, actual)
@ -39,7 +40,7 @@ class TestMartinM1(unittest.TestCase):
self.maxDiff = None
line_numbers = [1, 10, 100]
for line in line_numbers:
file = open("tests/assets/MartinM1_encode_line_lena%d.p" % line)
file = open(get_asset_filename("MartinM1_encode_line_lena%d.p" % line))
expected = pickle.load(file)
actual = list(self.lena.encode_line(line))
self.assertEqual(expected, actual)

9
pysstv/tests/test_sstv.py
View file

@ -6,8 +6,9 @@ from mock import MagicMock
from StringIO import StringIO
import hashlib
import sstv
from sstv import SSTV
from pysstv import sstv
from pysstv.sstv import SSTV
from common import get_asset_filename
class TestSSTV(unittest.TestCase):
@ -44,7 +45,7 @@ class TestSSTV(unittest.TestCase):
# 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("tests/assets/SSTV_gen_values.p"))
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)
@ -56,7 +57,7 @@ class TestSSTV(unittest.TestCase):
# 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("tests/assets/SSTV_gen_samples.p"))
expected = pickle.load(open(get_asset_filename("SSTV_gen_samples.p")))
actual = list(islice(gen_values, 0, 1000))
self.assertEqual(expected, actual)