diff options
| -rwxr-xr-x | Wrappers/Python/test/test_TestData.py | 107 |
1 files changed, 69 insertions, 38 deletions
diff --git a/Wrappers/Python/test/test_TestData.py b/Wrappers/Python/test/test_TestData.py index c4a0a70..fa0b98f 100755 --- a/Wrappers/Python/test/test_TestData.py +++ b/Wrappers/Python/test/test_TestData.py @@ -1,10 +1,9 @@ import numpy from ccpi.framework import TestData import os, sys - sys.path.append(os.path.dirname(os.path.abspath(__file__))) from testclass import CCPiTestClass - +import unittest class TestTestData(CCPiTestClass): def test_random_noise(self): @@ -17,67 +16,99 @@ class TestTestData(CCPiTestClass): self.assertAlmostEqual(norm, 48.881268, places=4) def test_load_CAMERA(self): - loader = TestData() - image = loader.load(TestData.CAMERA) - if image: - res = True - else: - res = False + loader = TestData() + res = False + try: + image = loader.load(TestData.CAMERA) + if (image.shape[0] == 512) and (image.shape[1] == 512): + res = True + else: + print("Image dimension mismatch") + except FileNotFoundError: + print("File not found") + except: + print("Failed to load file") self.assertTrue(res) + def test_load_BOAT(self): loader = TestData() - image = loader.load(TestData.BOAT) - - if image: - res = True - else: - res = False + res = False + try: + image = loader.load(TestData.BOAT) + if (image.shape[0] == 512) and (image.shape[1] == 512): + res = True + else: + print("Image dimension mismatch") + except FileNotFoundError: + print("File not found") + except: + print("Failed to load file") self.assertTrue(res) def test_load_PEPPERS(self): loader = TestData() - image = loader.load(TestData.PEPPERS) - - if image: - res = True - else: - res = False + res = False + try: + image = loader.load(TestData.PEPPERS) + if (image.shape[0] == 512) and (image.shape[1] == 512) and (image.shape[2] == 3): + res = True + else: + print("Image dimension mismatch") + except FileNotFoundError: + print("File not found") + except: + print("Failed to load file") self.assertTrue(res) def test_load_RESOLUTION_CHART(self): loader = TestData() - image = loader.load(TestData.RESOLUTION_CHART) - - if image: - res = True - else: - res = False + res = False + try: + image = loader.load(TestData.RESOLUTION_CHART) + if (image.shape[0] == 512) and (image.shape[1] == 512): + res = True + else: + print("Image dimension mismatch") + except FileNotFoundError: + print("File not found") + except: + print("Failed to load file") self.assertTrue(res) def test_load_SIMPLE_PHANTOM_2D(self): loader = TestData() - image = loader.load(TestData.SIMPLE_PHANTOM_2D) - - if image: - res = True - else: - res = False + res = False + try: + image = loader.load(TestData.SIMPLE_PHANTOM_2D) + if (image.shape[0] == 512) and (image.shape[1] == 512): + res = True + else: + print("Image dimension mismatch") + except FileNotFoundError: + print("File not found") + except: + print("Failed to load file") self.assertTrue(res) def test_load_SHAPES(self): loader = TestData() - image = loader.load(TestData.SHAPES) - - if image: - res = True - else: - res = False + res = False + try: + image = loader.load(TestData.SHAPES) + if (image.shape[0] == 200) and (image.shape[1] == 300): + res = True + else: + print("Image dimension mismatch") + except FileNotFoundError: + print("File not found") + except: + print("Failed to load file") self.assertTrue(res) |