72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
"""Contains tests for chain module"""
|
|
import unittest
|
|
|
|
from shenzhen_solitaire.board import NumberCard, Position
|
|
from shenzhen_solitaire.solver.board_actions import MoveAction, BunkerizeAction, GoalAction, HuaKillAction
|
|
from shenzhen_solitaire.solver import board_possibilities
|
|
from .boards import TEST_BOARD
|
|
|
|
|
|
class ChainTest(unittest.TestCase):
|
|
"""Tests the chain class"""
|
|
|
|
def test_sequence(self) -> None:
|
|
"""Tests a given sequence. Might break if I change the iterators"""
|
|
|
|
self.assertTrue(TEST_BOARD.check_correct())
|
|
sequence = [
|
|
MoveAction(
|
|
cards=[
|
|
NumberCard(
|
|
suit=NumberCard.Suit.Red,
|
|
number=7),
|
|
NumberCard(
|
|
suit=NumberCard.Suit.Green,
|
|
number=6)],
|
|
source_id=3,
|
|
destination_id=7),
|
|
BunkerizeAction(
|
|
card=NumberCard(
|
|
suit=NumberCard.Suit.Red,
|
|
number=6),
|
|
bunker_id=0,
|
|
field_id=2,
|
|
to_bunker=True),
|
|
GoalAction(
|
|
card=NumberCard(
|
|
suit=NumberCard.Suit.Green,
|
|
number=1),
|
|
source_id=2,
|
|
source_position=Position.Field),
|
|
MoveAction(
|
|
cards=[
|
|
NumberCard(
|
|
suit=NumberCard.Suit.Red,
|
|
number=4)],
|
|
source_id=2,
|
|
destination_id=5),
|
|
GoalAction(
|
|
card=NumberCard(
|
|
suit=NumberCard.Suit.Red,
|
|
number=1),
|
|
source_id=2,
|
|
source_position=Position.Field),
|
|
HuaKillAction(source_field_id=2),
|
|
MoveAction(
|
|
cards=[
|
|
NumberCard(
|
|
suit=NumberCard.Suit.Black, number=9),
|
|
NumberCard(suit=NumberCard.Suit.Red, number=8)],
|
|
source_id=6,
|
|
destination_id=2),
|
|
GoalAction(
|
|
card=NumberCard(
|
|
suit=NumberCard.Suit.Green, number=2),
|
|
source_id=6,
|
|
source_position=Position.Field)
|
|
]
|
|
for action in sequence:
|
|
step = list(board_possibilities.possible_actions(TEST_BOARD))
|
|
self.assertIn(action, step)
|
|
action.apply(TEST_BOARD)
|