Convert test_decompression_bomb.py to use unittest module

Tests/test_decompression_bomb.py

@@ -1,37 +1,40 @@
-from tester import *
+from helper import unittest, PillowTestCase, tearDownModule
 
 from PIL import Image
 
 test_file = "Images/lena.ppm"
 
 
-def test_no_warning_small_file():
+class TestDecompressionBomb(PillowTestCase):
-    # Implicit assert: no warning.
-    # A warning would cause a failure.
-    Image.open(test_file)
 
+    def test_no_warning_small_file(self):
+        # Implicit assert: no warning.
+        # A warning would cause a failure.
+        Image.open(test_file)
 
-def test_no_warning_no_limit():
+    def test_no_warning_no_limit(self):
-    # Arrange
+        # Arrange
-    # Turn limit off
+        # Turn limit off
-    Image.MAX_IMAGE_PIXELS = None
+        Image.MAX_IMAGE_PIXELS = None
-    assert_equal(Image.MAX_IMAGE_PIXELS, None)
+        self.assertEqual(Image.MAX_IMAGE_PIXELS, None)
 
-    # Act / Assert
+        # Act / Assert
-    # Implicit assert: no warning.
+        # Implicit assert: no warning.
-    # A warning would cause a failure.
+        # A warning would cause a failure.
-    Image.open(test_file)
+        Image.open(test_file)
 
+    def test_warning(self):
+        # Arrange
+        # Set limit to a low, easily testable value
+        Image.MAX_IMAGE_PIXELS = 10
+        self.assertEqual(Image.MAX_IMAGE_PIXELS, 10)
 
-def test_warning():
+        # Act / Assert
-    # Arrange
+        self.assert_warning(
-    # Set limit to a low, easily testable value
+            Image.DecompressionBombWarning,
-    Image.MAX_IMAGE_PIXELS = 10
+            lambda: Image.open(test_file))
-    assert_equal(Image.MAX_IMAGE_PIXELS, 10)
 
-    # Act / Assert
+if __name__ == '__main__':
-    assert_warning(
+    unittest.main()
-        DecompressionBombWarning,
-        lambda: Image.open(test_file))
 
 # End of file