예제 #1
0
def test_get_sequence():
    board = Board()
    board[0][0] = 'C'
    board[1][0] = 'C'

    ns_seq = board.get_sequence(0, 0, 3, 'NS')
    assert ns_seq == {
        'values': ['C', 'C', 'O'],
        'indices': [(0, 0), (1, 0), (2, 0)]
    }

    ew_seq = board.get_sequence(0, 0, 2, 'EW')
    assert ew_seq == {'values': ['C', 'O'], 'indices': [(0, 0), (0, 1)]}
예제 #2
0
def test_get_sequence_exc():
    board = Board()
    with pytest.raises(ValueError) as excinfo:
        board.get_sequence(0, 0, 2, 'SE')
    assert 'invalid direction' in str(excinfo.value)