From 410dd606f370360acb6b8137256e3261aaebae40 Mon Sep 17 00:00:00 2001 From: Anton Romanovich Date: Tue, 5 Feb 2013 10:40:58 +0600 Subject: Fix tests --- test.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/test.py b/test.py index a1d450d..6efbb68 100644 --- a/test.py +++ b/test.py @@ -12,7 +12,7 @@ class MagicTest(unittest.TestCase): filename = os.path.join(self.TESTDATA_DIR, filename) with open(filename, 'rb') as f: - value = m.from_buffer(f.read(1024)) + value = m.from_buffer(f.read()) self.assertEqual(value, expected_value) value = m.from_file(filename) @@ -53,9 +53,11 @@ class MagicTest(unittest.TestCase): self.assertRaises(IOError, m.from_file, 'nonexistent') self.assertRaises(magic.MagicException, magic.Magic, magic_file='nonexistent') - os.environ['MAGIC'] = '/nonexistent' - self.assertRaises(magic.MagicException, magic.Magic) - del os.environ['MAGIC'] + os.environ['MAGIC'] = 'nonexistent' + try: + self.assertRaises(magic.MagicException, magic.Magic) + finally: + del os.environ['MAGIC'] if __name__ == '__main__': -- cgit v1.2.1