26 lines
716 B
Python
26 lines
716 B
Python
"""
|
|
Tests for inventory.belt logic.
|
|
|
|
Covers: potion type detection, belt toggle keys.
|
|
"""
|
|
import pytest
|
|
import numpy as np
|
|
from logger import Logger
|
|
|
|
|
|
class TestInventoryBelt:
|
|
def setup_method(self):
|
|
Logger.init()
|
|
Logger.remove_file_logger()
|
|
|
|
def test_belt_toggle_keys_returns_list(self):
|
|
from inventory.belt import _belt_toggle_keys
|
|
keys = _belt_toggle_keys()
|
|
assert isinstance(keys, list)
|
|
assert len(keys) > 0
|
|
|
|
def test_cut_potion_img_returns_array(self):
|
|
from inventory.belt import _cut_potion_img
|
|
img = np.full((100, 100, 3), 255, dtype=np.uint8)
|
|
result = _cut_potion_img(img, 0, 0)
|
|
assert isinstance(result, np.ndarray) |