Fix a test for png's lightweight cleaning on corrupted files

1 files changed, 2 insertions, 3 deletions

diff --git a/tests/test_corrupted_files.py b/tests/test_corrupted_files.py
index 1d66556..7f8e223 100644
--- a/tests/test_corrupted_files.py
+++ b/tests/test_corrupted_files.py
@@ -194,10 +194,9 @@ class TestCorruptedFiles(unittest.TestCase):
         os.remove('./tests/data/clean.jpg')
 
     def test_png_lightweight(self):
-        return
         shutil.copy('./tests/data/dirty.torrent', './tests/data/clean.png')
-        p = images.PNGParser('./tests/data/clean.png')
-        self.assertTrue(p.remove_all())
+        with self.assertRaises(ValueError):
+            p = images.PNGParser('./tests/data/clean.png')
         os.remove('./tests/data/clean.png')
 
     def test_avi(self):