Tests: add bgra_to_rgb()

BoboTiG · BoboTiG · commit 48a88f3de7b0 · 2016-10-17T18:52:58.000+02:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -30,3 +30,8 @@ def sct(display):
 @pytest.fixture(scope='session')
 def is_travis():
     return 'TRAVIS' in environ
+
+
+@pytest.fixture(scope='session')
+def raw():
+    return open('tests/monitor-1024x768.raw', 'rb').read()
diff --git a/tests/monitor-1024x768.raw b/tests/monitor-1024x768.raw
diff --git a/tests/test_bgra_to_rgb.py b/tests/test_bgra_to_rgb.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+import pytest
+
+
+def test_bad_length(sct):
+    sct.width = 1024
+    sct.height = 768
+    with pytest.raises(ValueError):
+        sct.bgra_to_rgb(bytearray(b'789c626001000000ffff030000060005'))
+
+
+def test_good_bytes(sct, raw):
+    sct.width = 1024
+    sct.height = 768
+    image = sct.bgra_to_rgb(raw)
+    assert isinstance(image, bytes)
+
+
+def test_good_bytearray(sct, raw):
+    sct.width = 1024
+    sct.height = 768
+    image = sct.bgra_to_rgb(bytearray(raw))
+    assert isinstance(image, bytes)
