From 71acda4182f3f66e80c9d5bd7bc72513029d6f42 Mon Sep 17 00:00:00 2001
From: Mark Harfouche <mark.harfouche@gmail.com>
Date: Fri, 1 Nov 2024 22:13:52 -0400
Subject: [PATCH] Add tests for opencv python

---
 .github/workflows/ci.yml | 2 +-
 tests/test_jpg.py        | 8 ++++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 6447033..fc49494 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -75,7 +75,7 @@ jobs:
         shell: bash
         run: |
             python -m pip install --upgrade pip
-            pip install -U jupyter_packaging pytest simplejpeg pillow
+            pip install -U jupyter_packaging pytest simplejpeg pillow opencv-python-headless
             pip install .
             rm -rf ./jupyter_rfb ./build ./egg-info
       - name: Test with pytest
diff --git a/tests/test_jpg.py b/tests/test_jpg.py
index e5b0711..2f3f43f 100644
--- a/tests/test_jpg.py
+++ b/tests/test_jpg.py
@@ -8,6 +8,7 @@
     select_encoder,
     SimpleJpegEncoder,
     PillowJpegEncoder,
+    OpenCVJpegEncoder,
 )
 
 
@@ -40,6 +41,13 @@ def test_pillow_jpeg_encoder():
     _perform_error_checks(encoder)
 
 
+def test_opencv_jpeg_encoder():
+    """Test the opencv encoder."""
+    encoder = OpenCVJpegEncoder()
+    _perform_checks(encoder)
+    _perform_error_checks(encoder)
+
+
 def _perform_checks(encoder):
     # RGB
     im = get_random_im(100, 100, 3)