diff options
Diffstat (limited to 'tests/test_corrupted_files.py')
| -rw-r--r-- | tests/test_corrupted_files.py | 5 |
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 | |||
| 4 | import shutil | 4 | import shutil |
| 5 | import os | 5 | import os |
| 6 | 6 | ||
| 7 | from libmat2 import pdf, images, audio, office, parser_factory, torrent | 7 | from libmat2 import pdf, images, audio, office, parser_factory, torrent, harmless |
| 8 | 8 | ||
| 9 | 9 | ||
| 10 | class TestUnsupportedFiles(unittest.TestCase): | 10 | class 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): |
