import pytest
from pyball import PyBall
from pyball.models.config import SituationCode


@pytest.fixture(scope='module')
def test_situation_codes():
    pyball = PyBall()
    return pyball.get_situation_codes()


def test_get_situation_codes_returns_situation_codes(test_situation_codes):
    assert isinstance(test_situation_codes, list)
    assert isinstance(test_situation_codes[0], SituationCode)

import pytest
from pyball import PyBall
from pyball.models.config import SituationCode


@pytest.fixture(scope='module')
def test_situation_codes():
    pyball = PyBall()
    return pyball.get_situation_codes()


def test_get_situation_codes_returns_situation_codes(test_situation_codes):
    assert isinstance(test_situation_codes, list)
    assert isinstance(test_situation_codes[0], SituationCode)
