16 lines
435 B
Python
16 lines
435 B
Python
import cv2
|
|
import pytest
|
|
import screen
|
|
|
|
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
|