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"]