test_pattern.py

from typing import Tuple, List

from cubelang.cube import ICubeSide
from cubelang.orientation import Color
from cubelang.sides import Component
from cubelang.pattern import Pattern


class MockCubeSide(ICubeSide[None]):
    def __init__(self, colors: List[List[Color]]):
        super().__init__()
        self._colors = colors

    @property
    def rows(self) -> int:
        return len(self._colors)

    @property
    def columns(self) -> int:
        return len(self._colors[0])

    def __getitem__(self, item: Tuple[int, int]) -> Component[None]:
        i, j = item
        return Component(self._colors[i][j], None)

    def __setitem__(self, key: Tuple[int, int], value: Component[None]) -> None:
        raise NotImplementedError()


COLORS = [
    [Color.RED, Color.GREEN, Color.BLUE],
    [Color.WHITE, Color.WHITE, Color.ORANGE],
    [Color.ORANGE, Color.YELLOW, Color.YELLOW]
]


def test_pattern_match_correct():
    side = MockCubeSide(COLORS)
    pattern = Pattern([
        [None, Color.GREEN, None],
        ["g1", "g1", None],
        [None, "g2", "g2"]
    ])
    match = pattern.match(side, {"g1": Color.WHITE})
    assert match == {"g2": Color.YELLOW}


def test_pattern_match_wrong_color():
    side = MockCubeSide(COLORS)
    pattern = Pattern([
        [None, Color.RED, None],
        ["g1", "g1", None],
        [None, None, None]
    ])
    assert pattern.match(side, dict()) is None


def test_pattern_match_wrong_known():
    side = MockCubeSide(COLORS)
    pattern = Pattern([
        [None, Color.GREEN, None],
        ["g1", Color.WHITE, None],
        [None, "g2", "g2"]
    ])
    assert pattern.match(side, {"g1": Color.YELLOW}) is None


def test_pattern_match_wrong_found():
    side = MockCubeSide(COLORS)
    pattern = Pattern([
        [None, None, None],
        ["g1", None, None],
        ["g1", None, None]
    ])
    assert pattern.match(side, dict()) is None


def test_pattern_multiple_fount():
    side = MockCubeSide(COLORS)
    pattern = Pattern([
        [None, None, None],
        ["g1", "g2", None],
        [None, None, None]
    ])
    assert pattern.match(side, dict()) is None


def test_pattern_multiple():
    side = MockCubeSide(COLORS)
    pattern = Pattern([
        [None, None, None],
        ["g1", "g2", None],
        [None, None, None]
    ])
    assert pattern.match(side, {"g1": Color.WHITE}) is None