49 lines
966 B
Python
49 lines
966 B
Python
from ShipField import ShipField
|
|
|
|
|
|
def verify_value(actual, expected):
|
|
if actual == expected:
|
|
print("OK")
|
|
else:
|
|
print("ERROR")
|
|
|
|
|
|
my_field = ShipField()
|
|
|
|
print("set_ship_size()")
|
|
my_field.set_ship_size(1)
|
|
verify_value(my_field.set_ship_size, 1)
|
|
|
|
my_field.set_ship_size(0)
|
|
my_field.set_ship_size(6)
|
|
verify_value(my_field.set_ship_size, 1)
|
|
|
|
my_field.set_ship_size([])
|
|
my_field.set_ship_size("")
|
|
my_field.set_ship_size(None)
|
|
my_field.set_ship_size(True)
|
|
my_field.set_ship_size(False)
|
|
verify_value(my_field.set_ship_size, 1)
|
|
|
|
my_field.set_ship_size("2")
|
|
verify_value(my_field.set_ship_size, 2)
|
|
print()
|
|
|
|
print('toggle_field_mode()')
|
|
verify_value(my_field.field_mode, 0)
|
|
|
|
my_field.toggle_field_mode()
|
|
verify_value(my_field.field_mode, 1)
|
|
|
|
my_field.toggle_field_mode()
|
|
verify_value(my_field.field_mode, 0)
|
|
print()
|
|
|
|
print("ship_set()")
|
|
verify_value(my_field.field[0], " ")
|
|
|
|
my_field.set_ship_size(1)
|
|
my_field.ship_direction(0)
|
|
my_field.set_ship(0, 0)
|
|
|