test_subscriber.py

import pytest
from sorm.cdr import Subscriber, UserType

@pytest.fixture(name="Tetra_users")
def tetra_user():
    return [
        (Subscriber(UserType.inner, '102025000075780046ff', 501, 501), '102025000075780046'),
        (Subscriber(UserType.inner, '0e20250000755995ffffffffff', 65535, 65535), '0e20250000755995'),
        (Subscriber(UserType.inner, '102025000075782418ffffffffff', 506, 65535), '102025000075782418'),
    ]

         
@pytest.fixture(name="VSS_Users")
def vss_user():
    return [
        (Subscriber(UserType.outer, '62715810', 65535, 65535), '62715810'),
        (Subscriber(UserType.outer, '06733057ffffffffffffffffff', 65535, 65535), '06733057'),
        (Subscriber(UserType.outer, '793400', 65535, 65535), '793400')
    ]

@pytest.mark.parametrize("user,types", [(testra_user, "RASIO"), (vss_user, "VSS")])
def test_stuff(user, type, request):
    user_fixture = request.getfuncargvalue(user)
    pass

def test_radio_type(Tetra_users):
    for tupl in Tetra_users:
        user, _ = tupl
        assert user.get_type() == 'RADIO'


def test_vss_type(VSS_Users):
    for tupl in VSS_Users:
        user, _ = tupl
        assert user.get_type() == 'VSS'


def test_get_number(Tetra_users, VSS_Users):
    all_user = Tetra_users +  VSS_Users
    for tupl in all_user:
        user, number = tupl
        assert user.get_number() == number