test-random-emails.py

import random

import pytest

from script import get_random_emails


@pytest.fixture(autouse=True)
def set_random_seed():
    random.seed(123)


@pytest.fixture
def sample_emails():
    return [
        "[email protected]",
        "[email protected]",
        "[email protected]",
        "[email protected]",
        "[email protected]",
    ]


def test_get_random_emails(sample_emails):
    number_emails = 3
    result = get_random_emails(sample_emails, number_emails)
    expected_emails = ["[email protected]", "[email protected]", "[email protected]"]
    assert result == expected_emails