From e6b169bb4da4e2f8a2ef6309e13d30ac61425eb1 Mon Sep 17 00:00:00 2001
From: Muhammad Umair Shahid <58398786+Umair0343@users.noreply.github.com>
Date: Fri, 10 May 2024 01:50:50 +0500
Subject: [PATCH] Update test.py

Test cases added.
---
 tests/test_diff_motion_detector/test.py | 40 +++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/tests/test_diff_motion_detector/test.py b/tests/test_diff_motion_detector/test.py
index e468af8..063bfb0 100644
--- a/tests/test_diff_motion_detector/test.py
+++ b/tests/test_diff_motion_detector/test.py
@@ -2,14 +2,50 @@
 from deepgaze.deepgaze.motion_detection import DiffMotionDetector
 import cv2
 
-
 import pytest
 
 class TestDiffMotionDetector:
-
     # Tests that the background image is set successfully
     def test_set_background_successfully(self):
         detector = DiffMotionDetector()
         background_image = cv2.imread('background.jpg')
         detector.setBackground(background_image)
         assert detector.getBackground() is not None
+
+    # Tests that a binary image is returned successfully after the detection process
+    def test_return_binary_image_successfully(self):
+        detector = DiffMotionDetector()
+        background_image = cv2.imread('background.jpg')
+        foreground_image = cv2.imread('foreground.jpg')
+        detector.setBackground(background_image)
+        binary_image = detector.returnMask(foreground_image)
+        assert binary_image is not None
+
+    # Tests that a background image is set and a binary image is returned successfully after the detection process
+    def test_set_background_and_return_binary_image_successfully(self):
+        detector = DiffMotionDetector()
+        background_image = cv2.imread('background.jpg')
+        foreground_image = cv2.imread('foreground.jpg')
+        detector.setBackground(background_image)
+        binary_image = detector.returnMask(foreground_image)
+        assert detector.getBackground() is not None and binary_image is not None
+
+    # Tests that setting a None background image returns None
+    def test_set_none_background_image_and_return_none(self):
+        detector = DiffMotionDetector()
+        detector.setBackground(None)
+        assert detector.getBackground() is None
+
+    # Tests that setting a None foreground image returns None
+    def test_set_none_foreground_image_and_return_none(self):
+        detector = DiffMotionDetector()
+        background_image = cv2.imread('background.jpg')
+        detector.setBackground(background_image)
+        binary_image = detector.returnMask(None)
+        assert binary_image is None
+
+    # Tests that setting a None background image returns None after the detection process
+    def test_set_none_background_image_and_return_none_after_detection_process(self):
+        detector = DiffMotionDetector()
+        detector.setBackground(None)
+        assert detector.getBackground() is None