diff options
| -rwxr-xr-x | Wrappers/Python/test/test_TestData.py | 78 |
1 files changed, 71 insertions, 7 deletions
diff --git a/Wrappers/Python/test/test_TestData.py b/Wrappers/Python/test/test_TestData.py index 3b26612..c4a0a70 100755 --- a/Wrappers/Python/test/test_TestData.py +++ b/Wrappers/Python/test/test_TestData.py @@ -1,19 +1,83 @@ - import numpy from ccpi.framework import TestData import os, sys -sys.path.append( os.path.dirname( os.path.abspath(__file__) ) ) + +sys.path.append(os.path.dirname(os.path.abspath(__file__))) from testclass import CCPiTestClass class TestTestData(CCPiTestClass): def test_random_noise(self): - #loader = TestData(data_dir=os.path.join(sys.prefix, 'share','ccpi')) - #data_dir=os.path.join(os.path.dirname(__file__),'..', 'data') + # loader = TestData(data_dir=os.path.join(sys.prefix, 'share','ccpi')) + # data_dir=os.path.join(os.path.dirname(__file__),'..', 'data') loader = TestData() - camera = loader.load(TestData.CAMERA) - noisy_camera = TestData.random_noise(camera, seed=1) - norm = (camera-noisy_camera).norm() + norm = (camera - noisy_camera).norm() 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 + + self.assertTrue(res) + + def test_load_BOAT(self): + loader = TestData() + image = loader.load(TestData.BOAT) + + if image: + res = True + else: + res = False + + self.assertTrue(res) + + def test_load_PEPPERS(self): + loader = TestData() + image = loader.load(TestData.PEPPERS) + + if image: + res = True + else: + res = False + + self.assertTrue(res) + + def test_load_RESOLUTION_CHART(self): + loader = TestData() + image = loader.load(TestData.RESOLUTION_CHART) + + if image: + res = True + else: + res = False + + 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 + + self.assertTrue(res) + + def test_load_SHAPES(self): + loader = TestData() + image = loader.load(TestData.SHAPES) + + if image: + res = True + else: + res = False + + self.assertTrue(res) |