Parametrize PIL tests for DRYness

Author: akx

qrcode/tests/test_qrcode_pil.py

@@ -19,20 +19,14 @@ def test_render_pil():
     assert isinstance(img.get_image(), Image.Image)
 
 
-def test_render_pil_with_transparent_background():
+@pytest.mark.parametrize("back_color", ["TransParent", "red", (255, 195, 235)])
+def test_render_pil_background(back_color):
     qr = qrcode.QRCode()
     qr.add_data(UNICODE_TEXT)
     img = qr.make_image(back_color="TransParent")
     img.save(io.BytesIO())
 
 
-def test_render_pil_with_red_background():
-    qr = qrcode.QRCode()
-    qr.add_data(UNICODE_TEXT)
-    img = qr.make_image(back_color="red")
-    img.save(io.BytesIO())
-
-
 def test_render_pil_with_rgb_color_tuples():
     qr = qrcode.QRCode()
     qr.add_data(UNICODE_TEXT)
@@ -72,138 +66,49 @@ def test_render_styled_with_embeded_image_path(tmp_path):
     img.save(io.BytesIO())
 
 
-def test_render_styled_with_square_module_drawer():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    img = qr.make_image(
-        image_factory=StyledPilImage,
-        module_drawer=moduledrawers.SquareModuleDrawer(),
-    )
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_gapped_module_drawer():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    img = qr.make_image(
-        image_factory=StyledPilImage,
-        module_drawer=moduledrawers.GappedSquareModuleDrawer(),
-    )
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_circle_module_drawer():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    img = qr.make_image(
-        image_factory=StyledPilImage,
-        module_drawer=moduledrawers.CircleModuleDrawer(),
-    )
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_rounded_module_drawer():
+@pytest.mark.parametrize("drawer", [
+    moduledrawers.CircleModuleDrawer,
+    moduledrawers.GappedSquareModuleDrawer,
+    moduledrawers.HorizontalBarsDrawer,
+    moduledrawers.RoundedModuleDrawer,
+    moduledrawers.SquareModuleDrawer,
+    moduledrawers.VerticalBarsDrawer,
+])
+def test_render_styled_with_drawer(drawer):
     qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
     qr.add_data(UNICODE_TEXT)
     img = qr.make_image(
         image_factory=StyledPilImage,
-        module_drawer=moduledrawers.RoundedModuleDrawer(),
+        module_drawer=drawer(),
     )
     img.save(io.BytesIO())
 
 
-def test_render_styled_with_vertical_bars_module_drawer():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    img = qr.make_image(
-        image_factory=StyledPilImage,
-        module_drawer=moduledrawers.VerticalBarsDrawer(),
-    )
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_horizontal_bars_module_drawer():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    img = qr.make_image(
-        image_factory=StyledPilImage,
-        module_drawer=moduledrawers.HorizontalBarsDrawer(),
-    )
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_default_solid_color_mask():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.SolidFillColorMask()
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_solid_color_mask():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.SolidFillColorMask(back_color=WHITE, front_color=RED)
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_color_mask_with_transparency():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.SolidFillColorMask(
+@pytest.mark.parametrize("mask", [
+    colormasks.SolidFillColorMask(),
+    colormasks.SolidFillColorMask(back_color=WHITE, front_color=RED),
+    colormasks.SolidFillColorMask(
         back_color=(255, 0, 255, 255), front_color=RED
-    )
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-    assert img.mode == "RGBA"
-
-
-def test_render_styled_with_radial_gradient_color_mask():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.RadialGradiantColorMask(
+    ),
+    colormasks.RadialGradiantColorMask(
         back_color=WHITE, center_color=BLACK, edge_color=RED
-    )
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_square_gradient_color_mask():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.SquareGradiantColorMask(
+    ),
+    colormasks.SquareGradiantColorMask(
         back_color=WHITE, center_color=BLACK, edge_color=RED
-    )
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_horizontal_gradient_color_mask():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.HorizontalGradiantColorMask(
+    ),
+    colormasks.HorizontalGradiantColorMask(
         back_color=WHITE, left_color=RED, right_color=BLACK
-    )
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_vertical_gradient_color_mask():
-    qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
-    qr.add_data(UNICODE_TEXT)
-    mask = colormasks.VerticalGradiantColorMask(
+    ),
+    colormasks.VerticalGradiantColorMask(
         back_color=WHITE, top_color=RED, bottom_color=BLACK
-    )
-    img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
-    img.save(io.BytesIO())
-
-
-def test_render_styled_with_image_color_mask():
-    img_mask = Image.new("RGB", (10, 10), color="red")
+    ),
+    colormasks.ImageColorMask(
+        back_color=WHITE, color_mask_image=Image.new("RGB", (10, 10), color="red")
+    ),
+])
+def test_render_styled_with_mask(mask):
     qr = qrcode.QRCode(error_correction=qrcode.ERROR_CORRECT_L)
     qr.add_data(UNICODE_TEXT)
-    mask = colormasks.ImageColorMask(back_color=WHITE, color_mask_image=img_mask)
     img = qr.make_image(image_factory=StyledPilImage, color_mask=mask)
     img.save(io.BytesIO())