class test_saxs(unittest.TestCase): saxsPilatus = "1492/bsa_013_01.edf" maskFile = "1491/Pcon_01Apr_msk.edf" maskRef = "1490/bioSaxsMaskOnly.edf" maskDummy = "1488/bioSaxsMaskDummy.edf" poniFile = "1489/bioSaxs.poni" ai = None tmp_dir = os.environ.get("PYFAI_TEMPDIR", os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp")) def setUp(self): self.edfPilatus = UtilsTest.getimage(self.__class__.saxsPilatus) self.maskFile = UtilsTest.getimage(self.__class__.maskFile) self.poniFile = UtilsTest.getimage(self.__class__.poniFile) self.maskRef = UtilsTest.getimage(self.__class__.maskRef) self.maskDummy = UtilsTest.getimage(self.__class__.maskDummy) self.ai = AzimuthalIntegrator() self.ai.load(self.poniFile) if not os.path.isdir(self.tmp_dir): os.mkdir(self.tmp_dir) def test_mask(self): """test the generation of mask""" print self.edfPilatus data = fabio.open(self.edfPilatus).data mask = fabio.open(self.maskFile).data assert abs(self.ai.makeMask(data, mask=mask).astype(int) - fabio.open(self.maskRef).data).max() == 0 assert abs(self.ai.makeMask(data, mask=mask, dummy= -2, delta_dummy=1.1).astype(int) - fabio.open(self.maskDummy).data).max() == 0