56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
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)
|