diff --git a/Code/Mantid/Framework/PythonAPI/CMakeLists.txt b/Code/Mantid/Framework/PythonAPI/CMakeLists.txt index 3da81e778a9804877f71d940e719a22d05439c9b..3bd3f3ab808fef947aa5403c00355f073575bc8d 100644 --- a/Code/Mantid/Framework/PythonAPI/CMakeLists.txt +++ b/Code/Mantid/Framework/PythonAPI/CMakeLists.txt @@ -49,6 +49,7 @@ set ( TEST_PY_FILES test/ImportTest.py test/WorkspaceTests.py test/SettingsTest.py test/SANSWorkflowTest.py + test/SANSMaskCommands.py ) if(UNITY_BUILD) diff --git a/Code/Mantid/Framework/PythonAPI/test/SANSMaskCommands.py b/Code/Mantid/Framework/PythonAPI/test/SANSMaskCommands.py new file mode 100644 index 0000000000000000000000000000000000000000..cc4ed603ba4fde2e269cb926915ffff7fa14fb62 --- /dev/null +++ b/Code/Mantid/Framework/PythonAPI/test/SANSMaskCommands.py @@ -0,0 +1,36 @@ +import unittest +from mantidsimple import * +import ISISCommandInterface as ISIS + +class SANSMaskCommands(unittest.TestCase): + """ + Tests SANS workflow algorithms + """ + + def setUp(self): + + self.test_ws_name = "SANSMaskCommands_ws" + LoadEmptyInstrument('LOQ_Definition_20020226-.xml', self.test_ws_name) + self.test_ws = mantid[self.test_ws_name] + + ISIS.LOQ() + + self.assertEqual(self.test_ws.getNumberHistograms(), 17778) + self.assertTrue(self.test_ws.readY(0)[0] > 0) + + def test_single_spectrum(self): + """ + Checks the ISIS specfic mask command for spectra numbers + """ + spec_nums1 = 7341 + spec_nums2 = 17341 + + ISIS.Mask('mask S'+str(spec_nums1)) + ISIS.Mask('MASK S'+str(spec_nums2)) + + ISIS.ReductionSingleton().mask.execute(ISIS.ReductionSingleton(), self.test_ws_name) + self.assertEqual(self.test_ws.readY(spec_nums1-1)[0], 0) + self.assertEqual(self.test_ws.readY(spec_nums2-1)[0], 0) + +if __name__ == '__main__': + unittest.main() \ No newline at end of file