import cv2
import pytest
import screen

import utils.download_test_assets # downloads assets if they don't already exist, doesn't need to be called
from target_detect import get_visible_targets
import screen

@pytest.mark.parametrize("screen_path, expected_res", [
    ("test/assets/mobs.png", True),
    ("test/assets/mobs_no_green_or_blue.png", False),
])
def test_target_detect(screen_path, expected_res):
    screen.set_window_position(0, 0)
    img = cv2.imread(screen_path)
    assert bool(len(get_visible_targets(img))) == expected_res