diff --git a/tests/test_diff_motion_detector/background.jpg b/tests/test_diff_motion_detector/background.jpg new file mode 100644 index 0000000..8d00379 Binary files /dev/null and b/tests/test_diff_motion_detector/background.jpg differ diff --git a/tests/test_diff_motion_detector/foreground.jpg b/tests/test_diff_motion_detector/foreground.jpg new file mode 100644 index 0000000..8d00379 Binary files /dev/null and b/tests/test_diff_motion_detector/foreground.jpg differ diff --git a/tests/test_diff_motion_detector/test.py b/tests/test_diff_motion_detector/test.py new file mode 100644 index 0000000..e54d806 --- /dev/null +++ b/tests/test_diff_motion_detector/test.py @@ -0,0 +1,54 @@ + +# Generated by CodiumAI +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 \ No newline at end of file