added test for center of rotation finder

1 files changed, 21 insertions, 0 deletions

diff --git a/Wrappers/Python/test/test_DataProcessor.py b/Wrappers/Python/test/test_DataProcessor.py
index 1c1de3a..a5bd9c6 100755
--- a/Wrappers/Python/test/test_DataProcessor.py
+++ b/Wrappers/Python/test/test_DataProcessor.py
@@ -11,8 +11,29 @@ from timeit import default_timer as timer
 

 from ccpi.framework import AX, CastDataContainer, PixelByPixelDataProcessor

 

+from ccpi.io.reader import NexusReader

+from ccpi.processors import CenterOfRotationFinder

+import wget

+import os

+

 class TestDataProcessor(unittest.TestCase):

+    def setUp(self):

+        wget.download('https://github.com/DiamondLightSource/Savu/raw/master/test_data/data/24737_fd.nxs')

+        self.filename = '24737_fd.nxs'

+

+    def tearDown(self):

+        os.remove(self.filename)

 

+    def test_CenterOfRotation(self):

+        reader = NexusReader(self.filename)

+        ad = reader.get_acquisition_data_whole()

+        print (ad.geometry)

+        cf = CenterOfRotationFinder()

+        cf.set_input(ad)

+        print ("Center of rotation", cf.get_output())

+        

+        

+        

     def test_DataProcessorChaining(self):

         shape = (2,3,4,5)

         size = shape[0]