def test_find_pbsjob_multiple(self): jobs = [PbsJob('123', 'R', 'host1'), PbsJob('abc', 'Q', '')] self.assertEqual(jobs[0], hc._find_pbsjob('123', jobs)) self.assertEqual(jobs[1], hc._find_pbsjob('abc', jobs))
def test_find_pbsjob(self): expected = PbsJob('123', 'R', 'host') self.assertEqual(expected.jobid, hc._find_pbsjob('123', [PbsJob('123', 'R', 'host'), PbsJob('abc', 'Q', '')]).jobid) self.assertEqual(None, hc._find_pbsjob('xyz', [PbsJob('123', 'R', 'host'), PbsJob('abc', 'Q', '')]))