Added unittests for test data

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)