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)