Add a test for Screenshot.bgra

BoboTiG · BoboTiG · commit f1c71bb2b410 · 2018-03-17T16:16:49.000+01:00
diff --git a/CHANGELOG b/CHANGELOG
@@ -4,6 +4,7 @@ History:
 
 dev     2018/xx/xx
       - add PNG compression level control
+      - MSS: add Screenshot.bgra attribute
 
 3.1.2   2018/01/05
       - removed support for Python 3.3
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -5,6 +5,10 @@ base.py
 -------
 - Added ``MSSBase.compression_level`` to control the PNG compression level
 
+screenshot.py
+-------------
+- Added ``Screenshot.bgra`` to get BGRA bytes.
+
 tools.py
 --------
 - Changed signature of ``to_png(data, size, output=None)`` to ``to_png(data, size, level=6, output=None)``. ``level`` is the Zlib compression level.
diff --git a/tests/test_third_party.py b/tests/test_third_party.py
@@ -46,6 +46,19 @@ def test_pil(sct):
     assert os.path.isfile('box.png')
 
 
+@pytest.mark.skipif(
+    Image is None,
+    reason='PIL module not available.')
+def test_pil_bgra(sct):
+    width, height = 16, 16
+    box = {'top': 0, 'left': 0, 'width': width, 'height': height}
+    sct_img = sct.grab(box)
+
+    img = Image.frombytes('RGBX', sct_img.size, sct_img.bgra)
+    assert img.mode == 'RGBX'
+    assert img.size == sct_img.size
+
+
 @pytest.mark.skipif(
     Image is None,
     reason='PIL module not available.')
