[#7] test numpy interface even with cv2

NaturalHistoryMuseum · Dec 1, 2016 · 9f96b35 · 9f96b35
1 parent 661e3eb
commit 9f96b35
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/pyzbar/tests/test_decode.py b/pyzbar/tests/test_decode.py
@@ -3,6 +3,8 @@
 
 from pathlib import Path
 
+import numpy as np
+
 from PIL import Image
 
 try:
@@ -83,6 +85,11 @@ def test_empty(self):
         expected = []
         self.assertEqual(expected, res)
 
+    def test_decode_numpy(self):
+        "Read image using Pillow and convert to numpy.ndarray"
+        res = decode(np.asarray(self.code128))
+        self.assertEqual(self.EXPECTED_CODE128, res)
+
     @unittest.skipIf(cv2 is None, 'OpenCV not installed')
     def test_decode_opencv(self):
         "Read image using OpenCV"

diff --git a/requirements.pip b/requirements.pip
@@ -4,5 +4,6 @@ enum34==1.1.6; python_version == '2.7'
 # TODO How to specify OpenCV? 'cv2>=2.4.8'
 coveralls>=1.1
 nose>=1.3.4
+numpy>=1.8.2
 pathlib>=1.0.1; python_version == '2.7'
 Pillow>=3.2.0
diff --git a/setup.py b/setup.py
@@ -45,7 +45,8 @@ def readme():
     },
     'tests_require': [
         # TODO How to specify OpenCV? 'cv2>=2.4.8',
-        PILLOW
+        PILLOW,
+        'numpy>=1.8.2',
     ],
     'include_package_data': True,
     'classifiers': [