"""Unit tests for StatementPeriod model."""

from abacus_worksheet.models.statement_period import StatementPeriod


def test_get_statement_periods():
    """Test to get statement_period's."""
    statement_periods = StatementPeriod.get_statement_periods(['2023'])

    assert len(statement_periods) == 3
    assert statement_periods[0].statement_period_id == 1
    assert statement_periods[1].statement_period_id == 2
    assert statement_periods[2].statement_period_id == 3