def test_dprefix(self):
"""Test dprefix and usage in parseline and make_header"""
h = SubmitFilter([],[])
self.assertEqual(PBS_DIRECTIVE_PREFIX_DEFAULT, '#PBS',
msg='PBS_DIRECTIVE_PREFIX_DEFAULT %s' % PBS_DIRECTIVE_PREFIX_DEFAULT)
self.assertEqual(h.dprefix, PBS_DIRECTIVE_PREFIX_DEFAULT,
msg='PBS_DIRECTIVE_PREFIX_DEFAULT set as default dprefix')
# test parseline with default dprefix
self.assertEqual(h.parseline('regular line of text'), None, msg='non-header returns None')
self.assertEqual(h.parseline('# sldjldkfjkg'), [], msg='regular header returns empty list')
self.assertEqual(h.parseline(' '), [], msg='whitespace returns empty list')
res = h.parseline('#PBS -x y=z -a b=c,d=e,f,g # some comment')
self.assertEqual(h.parseline('#PBS -x y=z -a b=c,d=e,f,g # some comment'),
[('x', 'y=z'), ('a', 'b=c,d=e,f,g')], msg='PBS header with 2 options and comment (%s)' % (res,))
self.assertEqual(h.make_header('-a','b'), '#PBS -a b', 'Generate correct header')
# drpefix from enviornment
os.environ['PBS_DPREFIX'] = '#ENVDPREFIX'
h1 = SubmitFilter([],[])
self.assertEqual(h1.dprefix, '#ENVDPREFIX', msg='dprefix from environment wins')
# different dprefix from commandline
h2 = SubmitFilter(['-C', '#SOMETHINGELSE'], [])
self.assertEqual(h2.dprefix, '#SOMETHINGELSE', msg='cmdline dprefix wins')
# very different dprefix
self.assertEqual(h2.parseline('#PBS -x y=z -a b'), [], 'new dprefix, this is now just a comment')
self.assertEqual(h2.parseline('#SOMETHINGELSE -x y=z -a b'), [('x', 'y=z'), ('a','b')], 'new dprefix, this is now just a comment')
self.assertEqual(h2.make_header('-a','b'), '#SOMETHINGELSE -a b', 'Generate correct header with modified dprefix')
del os.environ['PBS_DPREFIX']
def test_dprefix(self):
"""Test dprefix and usage in parseline and make_header"""
h = SubmitFilter([], [])
self.assertEqual(PBS_DIRECTIVE_PREFIX_DEFAULT,
'#PBS',
msg='PBS_DIRECTIVE_PREFIX_DEFAULT %s' %
PBS_DIRECTIVE_PREFIX_DEFAULT)
self.assertEqual(
h.dprefix,
PBS_DIRECTIVE_PREFIX_DEFAULT,
msg='PBS_DIRECTIVE_PREFIX_DEFAULT set as default dprefix')
# test parseline with default dprefix
self.assertEqual(h.parseline('regular line of text'),
None,
msg='non-header returns None')
self.assertEqual(h.parseline('# sldjldkfjkg'), [],
msg='regular header returns empty list')
self.assertEqual(h.parseline(' '), [],
msg='whitespace returns empty list')
res = h.parseline('#PBS -x y=z -a b=c,d=e,f,g # some comment')
self.assertEqual(
h.parseline('#PBS -x y=z -a b=c,d=e,f,g # some comment'),
[('x', 'y=z'), ('a', 'b=c,d=e,f,g')],
msg='PBS header with 2 options and comment (%s)' % (res, ))
self.assertEqual(h.make_header('-a', 'b'), '#PBS -a b',
'Generate correct header')
# drpefix from enviornment
os.environ['PBS_DPREFIX'] = '#ENVDPREFIX'
h1 = SubmitFilter([], [])
self.assertEqual(h1.dprefix,
'#ENVDPREFIX',
msg='dprefix from environment wins')
# different dprefix from commandline
h2 = SubmitFilter(['-C', '#SOMETHINGELSE'], [])
self.assertEqual(h2.dprefix,
'#SOMETHINGELSE',
msg='cmdline dprefix wins')
# very different dprefix
self.assertEqual(h2.parseline('#PBS -x y=z -a b'), [],
'new dprefix, this is now just a comment')
self.assertEqual(h2.parseline('#SOMETHINGELSE -x y=z -a b'),
[('x', 'y=z'), ('a', 'b')],
'new dprefix, this is now just a comment')
self.assertEqual(h2.make_header('-a', 'b'), '#SOMETHINGELSE -a b',
'Generate correct header with modified dprefix')
del os.environ['PBS_DPREFIX']
def test_dprefix(self):
"""Test dprefix and usage in parseline and make_header"""
h = SubmitFilter([], [])
self.assertEqual(
PBS_DIRECTIVE_PREFIX_DEFAULT, "#PBS", msg="PBS_DIRECTIVE_PREFIX_DEFAULT %s" % PBS_DIRECTIVE_PREFIX_DEFAULT
)
self.assertEqual(
h.dprefix, PBS_DIRECTIVE_PREFIX_DEFAULT, msg="PBS_DIRECTIVE_PREFIX_DEFAULT set as default dprefix"
)
# test parseline with default dprefix
self.assertEqual(h.parseline("regular line of text"), None, msg="non-header returns None")
self.assertEqual(h.parseline("# sldjldkfjkg"), [], msg="regular header returns empty list")
self.assertEqual(h.parseline(" "), [], msg="whitespace returns empty list")
res = h.parseline("#PBS -x y=z -a b=c,d=e,f,g # some comment")
self.assertEqual(
h.parseline("#PBS -x y=z -a b=c,d=e,f,g # some comment"),
[("x", "y=z"), ("a", "b=c,d=e,f,g")],
msg="PBS header with 2 options and comment (%s)" % (res,),
)
self.assertEqual(h.make_header("-a", "b"), "#PBS -a b", "Generate correct header")
# drpefix from enviornment
os.environ["PBS_DPREFIX"] = "#ENVDPREFIX"
h1 = SubmitFilter([], [])
self.assertEqual(h1.dprefix, "#ENVDPREFIX", msg="dprefix from environment wins")
# different dprefix from commandline
h2 = SubmitFilter(["-C", "#SOMETHINGELSE"], [])
self.assertEqual(h2.dprefix, "#SOMETHINGELSE", msg="cmdline dprefix wins")
# very different dprefix
self.assertEqual(h2.parseline("#PBS -x y=z -a b"), [], "new dprefix, this is now just a comment")
self.assertEqual(
h2.parseline("#SOMETHINGELSE -x y=z -a b"),
[("x", "y=z"), ("a", "b")],
"new dprefix, this is now just a comment",
)
self.assertEqual(
h2.make_header("-a", "b"), "#SOMETHINGELSE -a b", "Generate correct header with modified dprefix"
)
del os.environ["PBS_DPREFIX"]
