import unittest

import cv2


class TestOpenCV(unittest.TestCase):
    def test_imread(self):
        img = cv2.imread('/input/tests/data/dot.png')

        self.assertEqual(1, img.shape[0])

    def test_classifier(self):
        face_classifier = cv2.CascadeClassifier()
        # Pretrained classifier downloaded from here: https://github.com/opencv/opencv/tree/4.x/data/haarcascades
        face_classifier.load('/input/tests/data/haarcascade_frontalface_alt.xml')

        img = cv2.imread('/input/tests/data/face.jpg')
        frame_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        frame_gray = cv2.equalizeHist(frame_gray)
        faces = face_classifier.detectMultiScale(frame_gray)
        self.assertEqual(1, len(faces))