test.py

import cv2
from app import black_and_white

def test_black_and_white():
    image = cv2.imread("test_image.jpg")
    result = black_and_white(image)
    assert result.shape == (image.shape[0], image.shape[1])