summaryrefslogtreecommitdiff
path: root/tests/test_corrupted_files.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_corrupted_files.py')
-rw-r--r--tests/test_corrupted_files.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/tests/test_corrupted_files.py b/tests/test_corrupted_files.py
index 4b2243d..1beb2ba 100644
--- a/tests/test_corrupted_files.py
+++ b/tests/test_corrupted_files.py
@@ -4,7 +4,7 @@ import unittest
4import shutil 4import shutil
5import os 5import os
6 6
7from libmat2 import pdf, images, audio, office, parser_factory, torrent 7from libmat2 import pdf, images, audio, office, parser_factory, torrent, harmless
8 8
9 9
10class TestUnsupportedFiles(unittest.TestCase): 10class TestUnsupportedFiles(unittest.TestCase):
@@ -65,8 +65,7 @@ class TestCorruptedFiles(unittest.TestCase):
65 65
66 def test_bmp(self): 66 def test_bmp(self):
67 shutil.copy('./tests/data/dirty.png', './tests/data/clean.bmp') 67 shutil.copy('./tests/data/dirty.png', './tests/data/clean.bmp')
68 with self.assertRaises(ValueError): 68 harmless.HarmlessParser('./tests/data/clean.bmp')
69 images.BMPParser('./tests/data/clean.bmp')
70 os.remove('./tests/data/clean.bmp') 69 os.remove('./tests/data/clean.bmp')
71 70
72 def test_docx(self): 71 def test_docx(self):