import IMP.test
import sys
import os
import re

class SAXSProfileApplicationTest(IMP.test.ApplicationTestCase):
    def test_simple(self):
        """Simple test of SAXS profile application"""
        print self.get_input_file_name('6lyz.pdb')
        print self.get_input_file_name('lyzexp.dat')
        p = self.run_application('foxs',
                                 [self.get_input_file_name('6lyz.pdb'),
                                  self.get_input_file_name('lyzexp.dat')])
        p.stdin.close()
        out = p.stdout.readlines()
        err = p.stderr.read()
        ret = p.wait()
        sys.stderr.write(err)
        self.assertApplicationExitedCleanly(ret)
        m = re.search('Chi\s+=\s+([\d\.]+)\r?', out[-1])
        self.assertNotEqual(m, None, msg="Chi output not found in " + str(out))
        self.assertAlmostEqual(float(m.group(1)), 0.44, delta=0.01)
        for out in ('6lyz.pdb.dat', '6lyz_lyzexp.dat', '6lyz_lyzexp.plt', '6lyz.plt'):
            os.unlink(self.get_input_file_name(out))

if __name__ == '__main__':
    IMP.test.main()