import random

TERRAIN_TYPES = [0, 1, 2, 3]


def generate_map(dimensions):
    x_len, y_len = dimensions
    return [[random.choice(TERRAIN_TYPES) for y in range(y_len)] for x in range(x_len)]