from src.utils.helpers import get_random_img


def test_get_random_img():
    img_names = ["img/1.png", "img/2.png", "img/3.png"]
    assert get_random_img(img_names) in img_names