def test_fail_invalid_infile(self):
'''Failing due to invalid input file.'''
tempname = common.get_temporary_filename(self.workroot)
nonexisting_infile = os.path.join(self.workdir, tempname)
outprefix = self.get_output('band')
cmdargs = [nonexisting_infile, outprefix]
with self.assertRaises(ScriptError):
dp_bands.main(cmdargs)
def test_no_enumeration(self):
'''Single spin channel and one k-point without enumeration'''
infile = self.get_input('band.out')
reffile = self.get_input('band_no-enumeration_tot.dat')
outfile = self.get_output('band_no-enumeration_tot.dat')
outprefix = self.get_output('band_no-enumeration')
cmdargs = ['-N', infile, outprefix]
dp_bands.main(cmdargs)
self.assertTrue(common.nxy_file_equals(outfile, reffile))
def test_two_spinchannels(self):
'''Two spin channels and multiple k-points'''
infile = self.get_input('band-separate-spins.out')
reffile_s1 = self.get_input('band-separate-spins_s1.dat')
reffile_s2 = self.get_input('band-separate-spins_s2.dat')
outfile_s1 = self.get_output('band-separate-spins_s1.dat')
outfile_s2 = self.get_output('band-separate-spins_s2.dat')
outprefix = self.get_output('band-separate-spins')
cmdargs = ['-s', infile, outprefix]
dp_bands.main(cmdargs)
self.assertTrue(common.nxy_file_equals(outfile_s1, reffile_s1))
self.assertTrue(common.nxy_file_equals(outfile_s2, reffile_s2))