/usr/lib/python2.7/dist-packages/PyMca/tests/specfilewrapperTest.py is in pymca 4.7.1+dfsg-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 | #/*##########################################################################
# Copyright (C) 2004-2012 European Synchrotron Radiation Facility
#
# This file is part of the PyMca X-ray Fluorescence Toolkit developed at
# the ESRF by the Software group.
#
# This file is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by the
# Free Software Foundation; either version 2 of the License, or (at your option)
# any later version.
#
# This file is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
# details.
#
#############################################################################*/
__author__ = "V.A. Sole - ESRF Data Analysis"
import unittest
import sys
import os
import gc
import tempfile
class testSpecfilewrapper(unittest.TestCase):
def setUp(self):
"""
import the module
"""
try:
from PyMca import specfilewrapper as specfile
self.specfileClass = specfile
except:
self.specfileClass = None
if self.specfileClass is not None:
text = "1.3 1 1\n"
text += "2.5 4 8\n"
text += "3.7 9 27\n"
text += "\n"
tmpFile = tempfile.mkstemp(text=False)
if sys.version < '3.0':
os.write(tmpFile[0], text)
else:
os.write(tmpFile[0], bytes(text, 'utf-8'))
os.close(tmpFile[0])
self.fname = tmpFile[1]
def tearDown(self):
"""clean up any possible files"""
# make sure the file handle is free
self._sf = None
self._scan = None
# this should free the handle
gc.collect()
if self.specfileClass is not None:
if os.path.exists(self.fname):
os.remove(self.fname)
def testSpecfilewrapperImport(self):
#"""Test successful import"""
self.assertTrue(self.specfileClass is not None,
'Unsuccessful PyMca.specfilewrapper import')
def testSpecfilewrapperReading(self):
#"""Test specfile readout"""
self.testSpecfilewrapperImport()
self._sf = self.specfileClass.Specfile(self.fname)
# test the number of found scans
self.assertEqual(len(self._sf), 2,
'Expected to read 2 scans, read %s' %\
len(self._sf))
self.assertEqual(self._sf.scanno(), 2,
'Expected to read 2 scans, got %s' %\
self._sf.scanno())
# test scan iteration selection method
self._scan = self._sf[0]
labels = self._scan.alllabels()
expectedLabels = ['Point', 'Column 0', 'Column 1', 'Column 2']
self.assertEqual(len(labels), 4,
'Expected to read 4 labels, got %s' % len(labels))
for i in range(3):
self.assertEqual(labels[i], expectedLabels[i],
'Read "%s" instead of "%s"' %\
(labels[i], expectedLabels[i]))
# test scan number selection method
self._scan = self._sf.select('1.1')
labels = self._scan.alllabels()
expectedLabels = ['Point', 'Column 0', 'Column 1', 'Column 2']
self.assertEqual(len(labels), 4,
'Expected to read 4 labels, got %s' % len(labels))
for i in range(3):
self.assertEqual(labels[i], expectedLabels[i],
'Read "%s" instead of "%s"' %\
(labels[i], expectedLabels[i]))
# test scan number of mca
self._scan = self._sf[0]
nbmca = self._scan.nbmca()
self.assertEqual(nbmca, 0,
'Expected to read 0 mca, got %s' % nbmca)
self._scan = self._sf[1]
nbmca = self._scan.nbmca()
self.assertEqual(nbmca, 3,
'Expected to read 3 mca, got %s' % nbmca)
def testSpecfilewrapperReadingCompatibleWithUserLocale(self):
#"""Test specfile compatible with C locale"""
self.testSpecfilewrapperImport()
self._sf = self.specfileClass.Specfile(self.fname)
self._scan = self._sf[0]
datacol = self._scan.datacol(2)
data = self._scan.data()
self._sf = None
self.assertEqual(datacol[0], 1.3,
'Read %f instead of %f' %\
(datacol[0], 1.3))
self.assertEqual(datacol[1], 2.5,
'Read %f instead of %f' %\
(datacol[1], 2.5))
self.assertEqual(datacol[2], 3.7,
'Read %f instead of %f' %\
(datacol[2], 3.7))
self.assertEqual(datacol[1], data[1][1],
'Read %f instead of %f' %\
(datacol[1], data[1][1]))
gc.collect()
def testTrainingSpectrumReading(self):
from PyMca import PyMcaDataDir
import numpy
fname = os.path.join(PyMcaDataDir.PYMCA_DATA_DIR,
'XRFSpectrum.mca')
self._sf = self.specfileClass.Specfile(fname)
self._scan = self._sf[0]
# I find awful that starts counting at 1
# 1 is the point number
# 2 is the actual spectal data
datacol = self._scan.datacol(2)
self._scan = self._sf[1]
# The "second" scan is the readout as mca
mca = self._scan.mca(1)
self.assertTrue(numpy.alltrue(datacol == mca))
def getSuite(auto=True):
testSuite = unittest.TestSuite()
if auto:
testSuite.addTest(\
unittest.TestLoader().loadTestsFromTestCase(testSpecfilewrapper))
else:
# use a predefined order
testSuite.addTest(testSpecfilewrapper("testSpecfilewrapperImport"))
testSuite.addTest(testSpecfilewrapper("testSpecfilewrapperReading"))
testSuite.addTest(\
testSpecfilewrapper(\
"testSpecfilewrapperReadingCompatibleWithUserLocale"))
testSuite.addTest(testSpecfilewrapper("testTrainingSpectrumReading"))
return testSuite
def test(auto=False):
unittest.TextTestRunner(verbosity=2).run(getSuite(auto=auto))
if __name__ == '__main__':
test()
|