#!/usr/bin/env python
# TestCCP4.py
#
#   Copyright (C) 2006 CCLRC, Graeme Winter
#
#   This code is distributed under the BSD license, a copy of which is 
#   included in the root directory of this package.
#
# 31st May 2006
# 
# Unit test suite for the CCP4 Driver interface - this will ensure that 
# everything is working properly for CCP4 examples. This makes use of 
# provided input data (two wavelengths worth).
#  
# 

import os, sys

if not os.environ.has_key('XIA2CORE_ROOT'):
    raise RuntimeError, 'XIA2CORE_ROOT not defined'

sys.path.append(os.path.join(os.environ['XIA2CORE_ROOT'],
                             'Python',
                             'Examples',
                             'CCP4'))

from Scala import Scala
from Sortmtz import Sortmtz

import unittest

class TestCCP4(unittest.TestCase):

    def setUp(self):
        pass

    def dosortinfl(self):
        '''Test sorting the inflection point data.'''

        s = Sortmtz()

        hklin = os.path.join(os.environ['XIA2CORE_ROOT'],
                             'Data', 'Test', 'Mtz', '12287_1_E1.mtz')

        hklout = '12287_1_E1_sorted.mtz'

        s = Sortmtz()
        
        s.set_hklin(hklin)
        s.set_hklout(hklout)
        
        status = s.sort()

        self.assertEqual(status, 'Normal termination')

        return

    def dosortlrem(self):
        '''Test sorting the low remote data.'''

        s = Sortmtz()

        hklin = os.path.join(os.environ['XIA2CORE_ROOT'],
                             'Data', 'Test', 'Mtz', '12287_1_E2.mtz')

        hklout = '12287_1_E2_sorted.mtz'

        s = Sortmtz()
        
        s.set_hklin(hklin)
        s.set_hklout(hklout)
        
        status = s.sort()

        self.assertEqual(status, 'Normal termination')

        return

    def doscaleinfl(self):
        '''Test scaling the inflection point data.'''

        s = Scala()

        hklin = '12287_1_E1_sorted.mtz'
        hklout = '12287_1_E1_scaled.broken'
        scalepack = '12287_1_E1_scaled.sca'

        s.set_hklin(hklin)
        s.set_hklout(hklout)
        s.set_scalepack(scalepack)

        status = s.scale()

        self.assertEqual(status, 'Normal termination')

    def doscalelrem(self):
        '''Test scaling the low remote data.'''

        s = Scala()

        hklin = '12287_1_E2_sorted.mtz'
        hklout = '12287_1_E2_scaled.broken'
        scalepack = '12287_1_E2_scaled.sca'

        s.set_hklin(hklin)
        s.set_hklout(hklout)
        s.set_scalepack(scalepack)

        status = s.scale()

        self.assertEqual(status, 'Normal termination')

    def testall(self):

        self.dosortinfl()
        self.dosortlrem()
        self.doscaleinfl()
        self.doscalelrem()

if __name__ == '__main__':
    unittest.main()