Skip to content

Commit 0e5b203

Browse files
committed
chore(test_image_sets_normalization): fix one series test
1 parent 71ec4d5 commit 0e5b203

File tree

1 file changed

+41
-27
lines changed

1 file changed

+41
-27
lines changed

packages/dicom/python/itkwasm-dicom-wasi/tests/test_image_sets_normalization.py

Lines changed: 41 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -3,38 +3,52 @@
33
from .common import test_input_path, test_output_path
44

55

6+
orientation_series = [
7+
test_input_path / "DicomImageOrientationTest" / "ImageOrientation.1.dcm",
8+
test_input_path / "DicomImageOrientationTest" / "ImageOrientation.2.dcm",
9+
test_input_path / "DicomImageOrientationTest" / "ImageOrientation.3.dcm",
10+
]
11+
12+
mr_series = [
13+
test_input_path / "dicom-images" / "MR" / "1-001.dcm",
14+
test_input_path / "dicom-images" / "MR" / "1-002.dcm",
15+
test_input_path / "dicom-images" / "MR" / "1-003.dcm",
16+
test_input_path / "dicom-images" / "MR" / "1-004.dcm",
17+
test_input_path / "dicom-images" / "MR" / "1-005.dcm",
18+
]
19+
20+
621
def test_one_series():
22+
assert orientation_series[0].exists()
23+
out_of_order = [
24+
orientation_series[1],
25+
orientation_series[2],
26+
orientation_series[0],
27+
]
28+
image_sets = image_sets_normalization(out_of_order)
29+
assert image_sets
30+
instances = list(image_sets[0]["Study"]["Series"].values())[0]["Instances"].values()
31+
sorted_files = [instance["FileName"] for instance in instances]
32+
assert all(
33+
file == str(path) for file, path in zip(sorted_files, orientation_series)
34+
)
35+
36+
37+
def test_two_series():
738
files = [
8-
test_input_path / "DicomImageOrientationTest" / "ImageOrientation.3.dcm",
9-
test_input_path / "DicomImageOrientationTest" / "ImageOrientation.1.dcm",
10-
test_input_path / "DicomImageOrientationTest" / "ImageOrientation.2.dcm",
39+
orientation_series[1],
40+
orientation_series[2],
41+
orientation_series[0],
42+
mr_series[3],
43+
mr_series[0],
44+
mr_series[4],
45+
mr_series[2],
46+
mr_series[1],
1147
]
1248
assert files[0].exists()
1349
image_sets = image_sets_normalization(files)
14-
print(image_sets)
15-
assert image_sets
16-
sorted_files = list(image_sets.values())[0]
17-
assert sorted_files == [
18-
str(files[1]),
19-
str(files[2]),
20-
str(files[0]),
21-
]
50+
assert len(image_sets) == 2
2251

23-
# def test_two_series():
24-
# files = [
25-
# test_input_path / "DicomImageOrientationTest" / "ImageOrientation.3.dcm",
26-
# test_input_path / "DicomImageOrientationTest" / "ImageOrientation.1.dcm",
27-
# test_input_path / "DicomImageOrientationTest" / "ImageOrientation.2.dcm",
28-
# test_input_path / "dicom-images" / "MR" / "1-001.dcm",
29-
# test_input_path / "dicom-images" / "MR" / "1-002.dcm",
30-
# test_input_path / "dicom-images" / "MR" / "1-003.dcm",
31-
# test_input_path / "dicom-images" / "MR" / "1-004.dcm",
32-
# test_input_path / "dicom-images" / "MR" / "1-005.dcm",
33-
# ]
34-
# assert files[0].exists()
35-
# image_sets = image_sets_normalization(files)
36-
# assert image_sets
37-
# assert len(image_sets.keys()) == 2
3852

3953
# def test_strange_ct():
4054
# files = [
@@ -43,4 +57,4 @@ def test_one_series():
4357
# ]
4458
# image_sets = image_sets_normalization(files)
4559
# assert image_sets
46-
# print(image_sets)
60+
# print(image_sets)

0 commit comments

Comments
 (0)