from __future__ import annotations

import pytest

from atlas.models import SeriesImage


@pytest.mark.django_db
def test_series_viewer_stack_description_includes_plane_and_contrast(create_series):
    series = create_series

    assert series.get_viewer_stack_description() == "Test Examination (Test Plane) / Test Contrast"
    assert series.get_series_dicom_json()["SeriesDescription"] == "Test Examination (Test Plane) / Test Contrast"