def test2(self): "multiphonon.getdos" from multiphonon.getdos import getDOS getDOS( os.path.join(datadir, "ARCS_V_annulus.nxs"), mt_nxs=os.path.join(datadir, "ARCS_V_annulus.nxs"), mt_fraction=0.01, ) return
def test1(self): "multiphonon.getdos" list(getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"))) self.assert_( close_hist( hh.load('work/final-dos.h5'), hh.load( os.path.join(here, 'expected_results', 'getdos-test1-final-dos.h5')))) return
def test1b(self): "multiphonon.getdos: reuse reduction results" work = 'work.getdos-reuse-reduction-results' if os.path.exists(work): import shutil shutil.rmtree(work) import warnings with warnings.catch_warnings(record=True) as ws: warnings.simplefilter('always') list( getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"), workdir=work)) for w in ws: self.assert_('Reusing old reduction' not in str(w)) continue # get dos again, this time we should see a warning with warnings.catch_warnings(record=True) as ws: warnings.simplefilter('always') list( getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"), workdir=work)) warned = False for w in ws: warned = warned or ('Reusing old reduction' in str(w)) if warned: break continue self.assert_(warned) # get dos using different settings. should not see warning with warnings.catch_warnings(record=True) as ws: warnings.simplefilter('always') list( getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"), Emin=0, workdir=work)) for w in ws: self.assert_('Reusing old reduction' not in str(w)) continue import histogram.hdf as hh iqe = hh.load(os.path.join(work, 'iqe.h5')) self.assert_(iqe.E[0] == 0.) return
def test1a(self): "multiphonon.getdos: check energy axis" import warnings with warnings.catch_warnings(record=True) as w: warnings.simplefilter('always') for _ in getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"), Emin=-50.5, Emax=80, dE=1.): assert len(w) == 1 assert "Energy axis modified" in str(w[-1].message) break return
def test3(self): "multiphonon.getdos: low T" list( getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"), T=1.5, workdir='work-lowT')) self.assert_( close_hist( hh.load('work-lowT/final-dos.h5'), hh.load( os.path.join(here, 'expected_results', 'getdos-test3-final-dos.h5')))) return
def test2(self): "multiphonon.getdos: MT can" list( getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs"), mt_nxs=os.path.join(datadir, "ARCS_V_annulus.nxs"), mt_fraction=0.01, workdir='work-MT')) self.assert_( close_hist( hh.load('work-MT/final-dos.h5'), hh.load( os.path.join(here, 'expected_results', 'getdos-test2-final-dos.h5')))) return
def test1(self): "multiphonon.getdos" from multiphonon.getdos import getDOS getDOS(os.path.join(datadir, "ARCS_V_annulus.nxs")) return