Source code for PyMca5.tests.DataTest

#/*##########################################################################
#
# The PyMca X-Ray Fluorescence Toolkit
#
# Copyright (c) 2004-2014 European Synchrotron Radiation Facility
#
# This file is part of the PyMca X-ray Fluorescence Toolkit developed at
# the ESRF by the Software group.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
#############################################################################*/
__author__ = "V. Armando Sole - ESRF Data Analysis"
__contact__ = "sole@esrf.fr"
__license__ = "MIT"
__copyright__ = "European Synchrotron Radiation Facility, Grenoble, France"
import unittest
import os
import sys
import numpy

[docs]class testData(unittest.TestCase): ELEMENTS = ['H', 'He', 'Li', 'Be', 'B', 'C', 'N', 'O', 'F', 'Ne', 'Na', 'Mg', 'Al', 'Si', 'P', 'S', 'Cl', 'Ar', 'K', 'Ca', 'Sc', 'Ti', 'V', 'Cr', 'Mn', 'Fe', 'Co', 'Ni', 'Cu', 'Zn', 'Ga', 'Ge', 'As', 'Se', 'Br', 'Kr', 'Rb', 'Sr', 'Y', 'Zr', 'Nb', 'Mo', 'Tc', 'Ru', 'Rh', 'Pd', 'Ag', 'Cd', 'In', 'Sn', 'Sb', 'Te', 'I', 'Xe', 'Cs', 'Ba', 'La', 'Ce', 'Pr', 'Nd', 'Pm', 'Sm', 'Eu', 'Gd', 'Tb', 'Dy', 'Ho', 'Er', 'Tm', 'Yb', 'Lu', 'Hf', 'Ta', 'W', 'Re', 'Os', 'Ir', 'Pt', 'Au', 'Hg', 'Tl', 'Pb', 'Bi', 'Po', 'At', 'Rn', 'Fr', 'Ra', 'Ac', 'Th', 'Pa', 'U', 'Np', 'Pu', 'Am', 'Cm', 'Bk', 'Cf', 'Es', 'Fm', 'Md', 'No', 'Lr', 'Rf', 'Db', 'Sg', 'Bh', 'Hs', 'Mt']
[docs] def setUp(self): """ Get the data directory """ self._importSuccess = False try: from PyMca5 import PyMcaDataDir self._importSuccess = True self.dataDir = PyMcaDataDir.PYMCA_DATA_DIR except: self.dataDir = None try: self.docDir = PyMcaDataDir.PYMCA_DOC_DIR except: self.docDir = None
[docs] def testDataDirectoryPresence(self): self.assertTrue(self._importSuccess, 'Unsuccessful PyMca5.PyMcaDataDir import') self.assertTrue(self.dataDir is not None, 'Unassigned PyMca5.PyMcaDataDir.PYMCA_DATA_DIR') self.assertTrue(os.path.exists(self.dataDir), 'Directory "%s" does not exist' % self.dataDir) self.assertTrue(os.path.isdir(self.dataDir), '"%s" expected to be a directory' % self.dataDir)
[docs] def testFisxDataDirectoryPresence(self): try: from fisx import DataDir dataDir = DataDir.FISX_DATA_DIR except: dataDir = None self.assertTrue(dataDir is not None, 'fisx module not properly installed') self.assertTrue(os.path.exists(dataDir), 'fisx directory "%s" does not exist' % dataDir) self.assertTrue(os.path.isdir(dataDir), '"%s" expected to be a directory' % dataDir) for fname in ['BindingEnergies.dat', 'EADL97_BindingEnergies.dat', 'EADL97_KShellConstants.dat', 'EADL97_LShellConstants.dat', 'EADL97_MShellConstants.dat', 'EPDL97_CrossSections.dat', 'KShellConstants.dat', 'KShellRates.dat', 'LShellConstants.dat', 'LShellRates.dat', 'MShellConstants.dat', 'MShellRates.dat', 'XCOM_CrossSections.dat']: actualName = os.path.join(dataDir, fname) self.assertTrue(os.path.exists(actualName), 'File "%s" does not exist.' % actualName) self.assertTrue(os.path.isfile(actualName), 'File "%s" is not an actual file.' % actualName)
[docs] def testDataFilePresence(self): # Testing file presence self.testDataDirectoryPresence() for fname in ['LShellRatesCampbell.dat', 'LShellRatesScofieldHS.dat', 'McaTheory.cfg', 'Scofield1973.dict', 'XRFSpectrum.mca']: actualName = os.path.join(self.dataDir, fname) self.assertTrue(os.path.exists(actualName), 'File "%s" does not exist.' % actualName) self.assertTrue(os.path.isfile(actualName), 'File "%s" is not an actual file.' % actualName) actualName = os.path.join(self.dataDir, 'attdata') self.assertTrue(os.path.exists(actualName), 'Directory "%s" does not exist' % actualName) self.assertTrue(os.path.isdir(actualName), '"%s" expected to be a directory' % actualName) for i in range(92): fname = "%s%s" % (testData.ELEMENTS[i], ".mat") actualName = os.path.join(self.dataDir, 'attdata', fname) self.assertTrue(os.path.exists(actualName), 'File "%s" does not exist.' % actualName) self.assertTrue(os.path.isfile(actualName), 'File "%s" is not an actual file.' % actualName)
[docs] def testDocDirectoryPresence(self): self.assertTrue(self._importSuccess, 'Unsuccessful PyMca5.PyMcaDataDir import') self.assertTrue(self.docDir is not None, 'Unassigned PyMca5.PyMcaDataDir.PYMCA_DOC_DIR') self.assertTrue(os.path.exists(self.dataDir), 'Directory "%s" does not exist' % self.docDir) self.assertTrue(os.path.isdir(self.dataDir), '"%s" expected to be a directory' % self.docDir)
[docs]def getSuite(auto=True): testSuite = unittest.TestSuite() if auto: testSuite.addTest(unittest.TestLoader().loadTestsFromTestCase(testData)) else: # use a predefined order testSuite.addTest(testData("testFisxDataDirectoryPresence")) testSuite.addTest(testData("testDataDirectoryPresence")) testSuite.addTest(testData("testDataFilePresence")) testSuite.addTest(testData("testDocDirectoryPresence")) return testSuite
[docs]def test(auto=False): unittest.TextTestRunner(verbosity=2).run(getSuite(auto=auto))
if __name__ == '__main__': test()