class test_azim_halfFrelon(unittest.TestCase):
"""basic test"""
fit2dFile = '1460/fit2d.dat'
halfFrelon = "1464/LaB6_0020.edf"
splineFile = "1461/halfccd.spline"
poniFile = "1463/LaB6.poni"
ai = None
fit2d = None
tmp_dir = os.environ.get("PYFAI_TEMPDIR", os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp"))
tmpfiles = {"cython":os.path.join(tmp_dir, "cython.dat"),
"cythonSP":os.path.join(tmp_dir, "cythonSP.dat"),
"numpy": os.path.join(tmp_dir, "numpy.dat")}
def setUp(self):
"""Download files"""
self.fit2dFile = UtilsTest.getimage(self.__class__.fit2dFile)
self.halfFrelon = UtilsTest.getimage(self.__class__.halfFrelon)
self.splineFile = UtilsTest.getimage(self.__class__.splineFile)
poniFile = UtilsTest.getimage(self.__class__.poniFile)
with open(poniFile) as f:
data = []
for line in f:
if line.startswith("SplineFile:"):
data.append("SplineFile: " + self.splineFile)
else:
data.append(line.strip())
self.poniFile = os.path.join(self.tmp_dir, os.path.basename(poniFile))
with open(self.poniFile, "w") as f:
f.write(os.linesep.join(data))
self.fit2d = numpy.loadtxt(self.fit2dFile)
self.ai = AzimuthalIntegrator()
self.ai.load(self.poniFile)
self.data = fabio.open(self.halfFrelon).data
if not os.path.isdir(self.tmp_dir):
os.makedirs(self.tmp_dir)
for tmpfile in self.tmpfiles.values():
if os.path.isfile(tmpfile):
os.unlink(tmpfile)
def tearDown(self):
"""Remove temporary files"""
unittest.TestCase.tearDown(self)
for tmpfile in self.tmpfiles.values():
if os.path.isfile(tmpfile):
os.unlink(tmpfile)
if os.path.isfile(self.poniFile):
os.unlink(self.poniFile)
def test_numpy_vs_fit2d(self):
"""
Compare numpy histogram with results of fit2d
"""
# logger.info(self.ai.__repr__())
tth, I = self.ai.xrpd_numpy(self.data,
len(self.fit2d), self.tmpfiles["numpy"], correctSolidAngle=False)
rwp = Rwp((tth, I), self.fit2d.T)
logger.info("Rwp numpy/fit2d = %.3f" % rwp)
if logger.getEffectiveLevel() == logging.DEBUG:
logger.info("Plotting results")
fig = pylab.figure()
fig.suptitle('Numpy Histogram vs Fit2D: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-b", label='fit2d')
sp.plot(tth, I, "-r", label="numpy histogram")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 11
def test_cython_vs_fit2d(self):
"""
Compare cython histogram with results of fit2d
"""
# logger.info(self.ai.__repr__())
tth, I = self.ai.xrpd_cython(self.data,
len(self.fit2d), "tmp/cython.dat", correctSolidAngle=False, pixelSize=None)
# logger.info(tth)
# logger.info(I)
rwp = Rwp((tth, I), self.fit2d.T)
logger.info("Rwp cython/fit2d = %.3f" % rwp)
if logger.getEffectiveLevel() == logging.DEBUG:
logger.info("Plotting results")
fig = pylab.figure()
fig.suptitle('Cython Histogram vs Fit2D: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-b", label='fit2d')
sp.plot(tth, I, "-r", label="cython")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 11
def test_cythonSP_vs_fit2d(self):
"""
Compare cython splitPixel with results of fit2d
"""
logger.info(self.ai.__repr__())
pos = self.ai.cornerArray(self.data.shape)
t0 = time.time()
logger.info("in test_cythonSP_vs_fit2d Before SP")
tth, I = self.ai.xrpd_splitPixel(self.data,
len(self.fit2d), self.tmpfiles["cythonSP"], correctSolidAngle=False)
logger.info("in test_cythonSP_vs_fit2d Before")
t1 = time.time() - t0
# logger.info(tth)
# logger.info(I)
rwp = Rwp((tth, I), self.fit2d.T)
logger.info("Rwp cythonSP(t=%.3fs)/fit2d = %.3f" % (t1, rwp))
if logger.getEffectiveLevel() == logging.DEBUG:
logger.info("Plotting results")
fig = pylab.figure()
fig.suptitle('CythonSP Histogram vs Fit2D: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-b", label='fit2d')
sp.plot(tth, I, "-r", label="cython")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 11
def test_cython_vs_numpy(self):
"""
Compare cython histogram with numpy histogram
"""
# logger.info(self.ai.__repr__())
data = self.data
tth_np, I_np = self.ai.xrpd_numpy(data,
len(self.fit2d), correctSolidAngle=False)
tth_cy, I_cy = self.ai.xrpd_cython(data,
len(self.fit2d), correctSolidAngle=False)
logger.info("before xrpd_splitPixel")
tth_sp, I_sp = self.ai.xrpd_splitPixel(data,
len(self.fit2d), correctSolidAngle=False)
logger.info("After xrpd_splitPixel")
rwp = Rwp((tth_cy, I_cy), (tth_np, I_np))
logger.info("Rwp = %.3f" % rwp)
if logger.getEffectiveLevel() == logging.DEBUG:
logging.info("Plotting results")
fig = pylab.figure()
fig.suptitle('Numpy Histogram vs Cython: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-y", label='fit2d')
sp.plot(tth_np, I_np, "-b", label='numpy')
sp.plot(tth_cy, I_cy , "-r", label="cython")
sp.plot(tth_sp, I_sp , "-g", label="SplitPixel")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 3
class test_azim_halfFrelon(unittest.TestCase):
"""basic test"""
fit2dFile = '1460/fit2d.dat'
halfFrelon = "1464/LaB6_0020.edf"
splineFile = "1461/halfccd.spline"
poniFile = "1463/LaB6.poni"
ai = None
fit2d = None
tmp_dir = os.environ.get("PYFAI_TEMPDIR", os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp"))
tmpfiles = {"cython":os.path.join(tmp_dir, "cython.dat"),
"cythonSP":os.path.join(tmp_dir, "cythonSP.dat"),
"numpy": os.path.join(tmp_dir, "numpy.dat")}
def setUp(self):
"""Download files"""
self.fit2dFile = UtilsTest.getimage(self.__class__.fit2dFile)
self.halfFrelon = UtilsTest.getimage(self.__class__.halfFrelon)
self.splineFile = UtilsTest.getimage(self.__class__.splineFile)
poniFile = UtilsTest.getimage(self.__class__.poniFile)
with open(poniFile) as f:
data = []
for line in f:
if line.startswith("SplineFile:"):
data.append("SplineFile: " + self.splineFile)
else:
data.append(line.strip())
self.poniFile = os.path.join(self.tmp_dir, os.path.basename(poniFile))
if not os.path.isdir(self.tmp_dir):
os.makedirs(self.tmp_dir)
with open(self.poniFile, "w") as f:
f.write(os.linesep.join(data))
self.fit2d = numpy.loadtxt(self.fit2dFile)
self.ai = AzimuthalIntegrator()
self.ai.load(self.poniFile)
self.data = fabio.open(self.halfFrelon).data
for tmpfile in self.tmpfiles.values():
if os.path.isfile(tmpfile):
os.unlink(tmpfile)
def tearDown(self):
"""Remove temporary files"""
unittest.TestCase.tearDown(self)
for tmpfile in self.tmpfiles.values():
if os.path.isfile(tmpfile):
os.unlink(tmpfile)
if os.path.isfile(self.poniFile):
os.unlink(self.poniFile)
def test_numpy_vs_fit2d(self):
"""
Compare numpy histogram with results of fit2d
"""
# logger.info(self.ai.__repr__())
tth, I = self.ai.xrpd_numpy(self.data,
len(self.fit2d), self.tmpfiles["numpy"], correctSolidAngle=False)
rwp = Rwp((tth, I), self.fit2d.T)
logger.info("Rwp numpy/fit2d = %.3f" % rwp)
if logger.getEffectiveLevel() == logging.DEBUG:
logger.info("Plotting results")
fig = pylab.figure()
fig.suptitle('Numpy Histogram vs Fit2D: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-b", label='fit2d')
sp.plot(tth, I, "-r", label="numpy histogram")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 11
def test_cython_vs_fit2d(self):
"""
Compare cython histogram with results of fit2d
"""
# logger.info(self.ai.__repr__())
tth, I = self.ai.xrpd_cython(self.data,
len(self.fit2d), self.tmpfiles["cython"], correctSolidAngle=False, pixelSize=None)
# logger.info(tth)
# logger.info(I)
rwp = Rwp((tth, I), self.fit2d.T)
logger.info("Rwp cython/fit2d = %.3f" % rwp)
if logger.getEffectiveLevel() == logging.DEBUG:
logger.info("Plotting results")
fig = pylab.figure()
fig.suptitle('Cython Histogram vs Fit2D: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-b", label='fit2d')
sp.plot(tth, I, "-r", label="cython")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 11
def test_cythonSP_vs_fit2d(self):
"""
Compare cython splitPixel with results of fit2d
"""
logger.info(self.ai.__repr__())
pos = self.ai.cornerArray(self.data.shape)
t0 = time.time()
logger.info("in test_cythonSP_vs_fit2d Before SP")
tth, I = self.ai.xrpd_splitPixel(self.data,
len(self.fit2d), self.tmpfiles["cythonSP"], correctSolidAngle=False)
logger.info("in test_cythonSP_vs_fit2d Before")
t1 = time.time() - t0
# logger.info(tth)
# logger.info(I)
rwp = Rwp((tth, I), self.fit2d.T)
logger.info("Rwp cythonSP(t=%.3fs)/fit2d = %.3f" % (t1, rwp))
if logger.getEffectiveLevel() == logging.DEBUG:
logger.info("Plotting results")
fig = pylab.figure()
fig.suptitle('CythonSP Histogram vs Fit2D: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-b", label='fit2d')
sp.plot(tth, I, "-r", label="cython")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 11
def test_cython_vs_numpy(self):
"""
Compare cython histogram with numpy histogram
"""
# logger.info(self.ai.__repr__())
data = self.data
tth_np, I_np = self.ai.xrpd_numpy(data,
len(self.fit2d), correctSolidAngle=False)
tth_cy, I_cy = self.ai.xrpd_cython(data,
len(self.fit2d), correctSolidAngle=False)
logger.info("before xrpd_splitPixel")
tth_sp, I_sp = self.ai.xrpd_splitPixel(data,
len(self.fit2d), correctSolidAngle=False)
logger.info("After xrpd_splitPixel")
rwp = Rwp((tth_cy, I_cy), (tth_np, I_np))
logger.info("Rwp = %.3f" % rwp)
if logger.getEffectiveLevel() == logging.DEBUG:
logging.info("Plotting results")
fig = pylab.figure()
fig.suptitle('Numpy Histogram vs Cython: Rwp=%.3f' % rwp)
sp = fig.add_subplot(111)
sp.plot(self.fit2d.T[0], self.fit2d.T[1], "-y", label='fit2d')
sp.plot(tth_np, I_np, "-b", label='numpy')
sp.plot(tth_cy, I_cy , "-r", label="cython")
sp.plot(tth_sp, I_sp , "-g", label="SplitPixel")
handles, labels = sp.get_legend_handles_labels()
fig.legend(handles, labels)
fig.show()
raw_input("Press enter to quit")
assert rwp < 3
