dtos-manage-breast-screening/manage_breast_screening/dicom/tests/test_models.py at efbf5cc18033395a613f16b9c151bdce23ee9d8a · NHSDigital/dtos-manage-breast-screening · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import pytest

from manage_breast_screening.dicom.tests.factories import (
    ImageFactory,
    SeriesFactory,
    StudyFactory,
)


@pytest.mark.django_db
class TestStudy:
    def test_study_has_series_with_multiple_images(self):
        series = SeriesFactory.create()
        ImageFactory.create_batch(2, series=series)
        study = series.study

        assert study.has_series_with_multiple_images() is True

    def test_study_series_with_multiple_images_only_counts_series_with_more_than_one_image(
        self,
    ):
        series = SeriesFactory.create()
        ImageFactory.create_batch(2, series=series)
        ImageFactory.create(series=series)
        study = series.study

        assert study.series_with_multiple_images().count() == 1

    def test_study_does_not_have_series_with_multiple_images(self):
        study = StudyFactory.create()
        ImageFactory.create(series__study=study)

        assert study.has_series_with_multiple_images() is False


@pytest.mark.django_db
class TestSeries:
    def test_series_count_property(self):
        series = SeriesFactory.create()
        ImageFactory.create_batch(3, series=series)

        assert series.count == 3

    def test_series_extra_count_property(self):
        series = SeriesFactory.create()
        ImageFactory.create_batch(3, series=series)

        assert series.extra_count == 2

    def test_series_count_property_no_images(self):
        series = SeriesFactory.create()

        assert series.count == 0

    def test_series_extra_count_property_implant_images(self):
        series = SeriesFactory.create()
        ImageFactory.create_batch(2, series=series, implant_present=True)

        assert series.extra_count == 0

    def test_series_laterality(self):
        series = SeriesFactory.create()
        ImageFactory.create(series=series, laterality="R")

        assert series.laterality == "R"

    def test_series_laterality_no_images(self):
        series = SeriesFactory.create()

        assert series.laterality == ""

    def test_series_view_position(self):
        series = SeriesFactory.create()
        ImageFactory.create(series=series, view_position="MLO")

        assert series.view_position == "MLO"

    def test_series_view_position_no_images(self):
        series = SeriesFactory.create()

        assert series.view_position == ""


@pytest.mark.django_db
class TestImage:
    @pytest.mark.parametrize(
        "laterality, view_position, expected",
        [
            ("R", "MLO", "RMLO"),
            ("R", "CC", "RCC"),
            ("L", "MLO", "LMLO"),
            ("L", "CC", "LCC"),
            ("R", "", ""),
            ("", "MLO", ""),
            ("", "", ""),
        ],
    )
    def test_laterality_and_view(self, laterality, view_position, expected):
        image = ImageFactory.build(laterality=laterality, view_position=view_position)

        assert image.laterality_and_view == expected

    def test_image_str_representation(self):
        image = ImageFactory.build(laterality="R", view_position="MLO")
        assert str(image) == "RMLO"