diff --git a/tests/test_mapa.py b/tests/test_mapa.py index 8c53b3a..5d1acb3 100644 --- a/tests/test_mapa.py +++ b/tests/test_mapa.py @@ -1,5 +1,7 @@ import pytest -from modules.mapa import Mapa +from random import randint +from modules.agente import Agente +from modules.mapa import Mapa, MAX_DIMENSAO, MIN_DIMENSAO class Test_Mapa: @@ -18,3 +20,21 @@ class Test_Mapa: with pytest.raises(ValueError) as e_info: mapa = Mapa((-1, -1)) print(e_info, type(mapa)) + + def test_posicao(self): + """ + testar retorno de posicao + """ + mapa: Mapa = Mapa((16, 16)) + pos_x: int = randint(MIN_DIMENSAO - 1, mapa.dimensao[1] + 1) + pos_y: int = randint(MIN_DIMENSAO - 1, mapa.dimensao[1] + 1) + agente: Agente = mapa.posicao((pos_x, pos_y)) + if pos_x < 0: + pos_x = mapa.dimensao[0] - 1 + if pos_x >= mapa.dimensao[0]: + pos_x = 0 + if pos_y < 0: + pos_y = mapa.dimensao[1] - 1 + if pos_y >= mapa.dimensao[0]: + pos_y = 0 + assert agente is mapa.mundo[pos_x][pos_y]