From ea8be660b90c0a35db0caa297435c597b09eaa15 Mon Sep 17 00:00:00 2001 From: Joseph Paul Cohen Date: Wed, 20 May 2020 12:00:36 -0400 Subject: [PATCH 1/2] add resize test --- tests/test_dataloaders.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/test_dataloaders.py b/tests/test_dataloaders.py index 5e2a701..01df89b 100644 --- a/tests/test_dataloaders.py +++ b/tests/test_dataloaders.py @@ -1,5 +1,6 @@ import pytest import torchxrayvision as xrv +from skimage.io import imread, imsave dataset_classes = [xrv.datasets.NIH_Dataset, xrv.datasets.PC_Dataset, @@ -45,3 +46,21 @@ def test_dataloader_merging_incorrect_alignment(): assert "incorrect pathology alignment" in str(excinfo.value) + +def test_dataloader_merging(): + + for filename in ["16747_3_1.jpg", "covid-19-pneumonia-58-prior.jpg"] + img = imread(filename) + img = xrv.datasets.normalize(img, 255) + + # Check that images are 2D arrays + if len(img.shape) > 2: + img = img[:, :, 0] + + # Add color channel + img = img[None, :, :] + + resize_ski = xrv.datasets.XRayResizer(100, engine="skimage") + resize_cv2 = xrv.datasets.XRayResizer(100, engine="cv2") + + assert(np.allclose(resize_ski(img),resize_cv2(img))) From 2fa9ef89039241267b8edba36f76fa63f3100071 Mon Sep 17 00:00:00 2001 From: Joseph Paul Cohen Date: Wed, 20 May 2020 12:02:43 -0400 Subject: [PATCH 2/2] fix function name --- tests/test_dataloaders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_dataloaders.py b/tests/test_dataloaders.py index 01df89b..dc2ce62 100644 --- a/tests/test_dataloaders.py +++ b/tests/test_dataloaders.py @@ -47,7 +47,7 @@ def test_dataloader_merging_incorrect_alignment(): assert "incorrect pathology alignment" in str(excinfo.value) -def test_dataloader_merging(): +def test_resize(): for filename in ["16747_3_1.jpg", "covid-19-pneumonia-58-prior.jpg"] img = imread(filename)