27 lines
1004 B
Python
27 lines
1004 B
Python
import pytest
|
|
import numpy as np
|
|
from logger import Logger
|
|
from game_stats import GameStats
|
|
|
|
|
|
class TestGameStats:
|
|
def setup_method(self):
|
|
Logger.init()
|
|
Logger.remove_file_logger()
|
|
self.game_stats = GameStats()
|
|
|
|
@pytest.mark.parametrize("item_name, item_should_be_added", [
|
|
("MAGIC ITEM", True),
|
|
("JEWEL", True),
|
|
("GREATER HEALING POTION", False),
|
|
("CHIPPED SKULL", False),
|
|
("FLAWLESS SKULL", False),
|
|
])
|
|
def test_log_item_keep(self, item_name: str, item_should_be_added: bool):
|
|
self.game_stats.update_location("test_location")
|
|
previous_item_count = len(self.game_stats._location_stats["test_location"]["items"])
|
|
self.game_stats.log_item_keep(item_name, send_message=False, img=np.zeros([5, 5]))
|
|
new_item_count = len(self.game_stats._location_stats["test_location"]["items"])
|
|
item_was_added = previous_item_count < new_item_count
|
|
assert(item_was_added == item_should_be_added)
|