ShipPos.py

import random

ship_positions = [
    [
        [(2, 5), (2, 6)],
        [(3, 7), (4, 7), (5, 7), (6, 7), (7, 7)],
        [(8, 5), (8, 6)],
        [(4, 2), (4, 3), (4, 4)],
        [(6, 2), (6, 3), (6, 4)]
    ],
    [
        [(7, 1), (7, 2), (7, 3)],
        [(4, 5), (5, 5), (6, 5), (7, 5), (8, 5)],
        [(4, 6), (4, 7)],
        [(5, 1), (5, 2), (5, 3)],
        [(8, 6), (8, 7)]
    ],
    [
        [(2, 1), (2, 3), (2, 4)],
        [(2, 7), (3, 7), (4, 7), (5, 7), (6, 7)],
        [(7, 7), (8, 7)],
        [(4, 2), (5, 2), (6, 2)],
        [(5, 9), (6, 9)]
    ],
    [
        [(1, 1), (2, 1), (3, 1)],
        [(9, 9), (9, 8), (9, 7), (9, 6), (9, 5)],
        [(7, 7), (8, 7)],
        [(3, 4), (4, 4), (5, 4)],
        [(6, 2), (6, 3)]
    ],
    [
        [(3, 1), (4, 1), (5, 1)],
        [(8, 1), (8, 2), (8, 3)],
        [(1, 4), (1, 5), (1, 6), (1, 7), (1, 8)],
        [(5, 7), (5, 8)],
        [(6, 8), (7, 8)]
    ],
    [
        [(6, 1), (6, 2), (6, 3), (6, 4), (6, 5)],
        [(3, 3), (3, 4), (3, 5)],
        [(8, 3), (9, 3)],
        [(6, 9), (7, 9), (8, 9)],
        [(2, 7), (2, 8)]
    ],
    [
        [(1, 1), (1, 2), (1, 3)],
        [(4, 2), (5, 2), (6, 2), (7, 2), (8, 2)],
        [(5, 4), (5, 5), (5, 6)],
        [(1, 8), (2, 8)],
        [(7, 9), (8, 9)]
    ],
    [
        [(1, 1), (2, 1), (3, 1)],
        [(3, 3), (3, 4), (3, 5), (3, 6), (3, 7)],
        [(8, 1), (8, 2)],
        [(4, 5), (5, 5), (6, 5)],
        [(5, 9), (6, 9)]
    ]
]

def getShipPos():
    """
    THIS IS THE LIST OF SHIPS
    [5,3,3,2,2]
    That is:
    1x 5 long
    2x 3 long
    2x 2 long

    Your ships must satisfy this
    """

    return ship_positions[random.randint(0, len(ship_positions) - 1)]