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 [
        "email1@example.com",
        "email2@example.com",
        "email3@example.com",
        "email4@example.com",
        "email5@example.com",
    ]


def test_get_random_emails(sample_emails):
    number_emails = 3
    result = get_random_emails(sample_emails, number_emails)
    expected_emails = ["email1@example.com", "email3@example.com", "email5@example.com"]
    assert result == expected_emails