From ac0f89bfdf9bf64e368de2b608821248114f915d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A1s=20Veres-Szentkir=C3=A1lyi?= Date: Thu, 12 Jan 2017 12:25:34 +0100 Subject: [PATCH] test_sstv: open further files in explicit 'rb' mode --- pysstv/tests/test_sstv.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pysstv/tests/test_sstv.py b/pysstv/tests/test_sstv.py index 9991996..10162c5 100644 --- a/pysstv/tests/test_sstv.py +++ b/pysstv/tests/test_sstv.py @@ -46,7 +46,8 @@ 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(get_asset_filename("SSTV_gen_values.p"))) + with open(get_asset_filename("SSTV_gen_values.p"), 'rb') as f: + expected = pickle.load(f) for e, g in zip(expected, gen_values): self.assertAlmostEqual(e, g, delta=0.000000001) @@ -58,7 +59,8 @@ 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(get_asset_filename("SSTV_gen_samples.p"))) + with open(get_asset_filename("SSTV_gen_samples.p"), 'rb') as f: + expected = pickle.load(f) actual = list(islice(gen_values, 0, 1000)) for e, a in zip(expected, actual): self.assertAlmostEqual(e, a, delta=1)