from random import randrange

import numpy as np
import torch

from src.utils import seed_everything


def test_reproducibility():
    seed_everything()
    python_random_a = randrange(100)
    np_random_a = np.random.randint(100, size=1)[0]
    torch_random_a = torch.randint(100, size=(1, ))[0]

    seed_everything()
    python_random_b = randrange(100)
    np_random_b = np.random.randint(100, size=1)[0]
    torch_random_b = torch.randint(100, size=(1, ))[0]

    assert python_random_a == python_random_b
    assert np_random_a == np_random_b
    assert torch_random_a == torch_random_b