|
|
import unittest |
|
|
|
|
|
from wrdler.logic import build_letter_map, reveal_cell, guess_word, is_game_over |
|
|
from wrdler.models import Coord, Word, Puzzle, GameState |
|
|
|
|
|
|
|
|
class TestLogic(unittest.TestCase): |
|
|
def make_state(self): |
|
|
w1 = Word("TREE", Coord(0, 0), "H") |
|
|
w2 = Word("APPLE", Coord(2, 0), "H") |
|
|
w3 = Word("ORANGE", Coord(4, 0), "H") |
|
|
w4 = Word("WIND", Coord(0, 6), "V") |
|
|
w5 = Word("MOUSE", Coord(0, 8), "V") |
|
|
w6 = Word("PYTHON", Coord(0, 10), "V") |
|
|
p = Puzzle([w1, w2, w3, w4, w5, w6]) |
|
|
state = GameState( |
|
|
grid_size=12, |
|
|
puzzle=p, |
|
|
revealed=set(), |
|
|
guessed=set(), |
|
|
score=0, |
|
|
last_action="", |
|
|
can_guess=False, |
|
|
) |
|
|
return state, p |
|
|
|
|
|
def test_reveal_and_guess_gating(self): |
|
|
state, puzzle = self.make_state() |
|
|
letter_map = build_letter_map(puzzle) |
|
|
|
|
|
ok, pts = guess_word(state, "TREE") |
|
|
self.assertFalse(ok) |
|
|
self.assertEqual(pts, 0) |
|
|
|
|
|
reveal_cell(state, letter_map, Coord(0, 0)) |
|
|
self.assertTrue(state.can_guess) |
|
|
ok, pts = guess_word(state, "TREE") |
|
|
self.assertTrue(ok) |
|
|
self.assertGreater(pts, 0) |
|
|
self.assertIn("TREE", state.guessed) |
|
|
self.assertFalse(state.can_guess) |
|
|
|
|
|
def test_game_over(self): |
|
|
state, puzzle = self.make_state() |
|
|
letter_map = build_letter_map(puzzle) |
|
|
|
|
|
for w in puzzle.words: |
|
|
reveal_cell(state, letter_map, w.start) |
|
|
ok, _ = guess_word(state, w.text) |
|
|
self.assertTrue(ok) |
|
|
self.assertTrue(is_game_over(state)) |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
unittest.main() |
