Madra_smieciarka/QOLfunc.py

# Tutaj wrzucam jakieś funkcje różne, co by nie zaśmiecać tym maina jak nie trzeba
import random


def chooseNeighbours(count: int, i: int) -> set:
    """
    :param count: ile sąsiadów ma wybrać
    :param i: id obiektu, żeby nie był sąsiadem sam ze sobą
    :return: zbiór id wybranych losowo sąsiadów obiektu
    """
    neigh = set()
    while len(neigh) < count:
        num = random.randint(0, 7)
        if num != i:
            neigh.add(num)
    return neigh