from battleship.utils import check_field


def test_simple():
    field = [
        ['o', 'o', 'o', 'o', ' ', 'o', 'o', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o'],
    ]
    field = [''.join(a) for a in field]
    field = ''.join(field)
    assert check_field(field)


def test_incorrect():
    field = [
        ['o', ' ', 'o', 'o', ' ', 'o', 'o', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o'],
    ]
    field = [''.join(a) for a in field]
    field = ''.join(field)
    assert not check_field(field)


def test_incorrect2():
    field = [
        ['o', 'o', 'o', 'o', ' ', 'o', 'o', 'o', ' ', 'o'],
        [' ', ' ', 'o', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', 'o', ' ', ' ', 'o', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', ' '],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o'],
        [' ', ' ', ' ', ' ', ' ', ' ', ' ', 'o', ' ', 'o'],
    ]
    field = [''.join(a) for a in field]
    field = ''.join(field)
    assert not check_field(field)