def testWorkLeft(self):
quests = [
quest_test.QuestCycle(quest_test.QuestPass(), quest_test.QuestSpin())
]
state = job_state.JobState(quests)
state.AddChange(change_test.Change(123))
self.assertTrue(state.ScheduleWork())
def testNoFailedAttempts(self):
quests = [quest_test.QuestPass()]
state = job_state.JobState(quests)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(2))
state.ScheduleWork()
self.assertFalse(state.FirstOrLastChangeFailed())
def testUnknown(self):
quests = [quest_test.QuestPass()]
state = job_state.JobState(quests,
comparison_mode=job_state.FUNCTIONAL,
comparison_magnitude=0.2)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(9))
state.ScheduleWork()
state.Explore()
# Need more information. Add more attempts to one Change.
self.assertEqual(len(state._changes), 2)
attempt_count_1 = len(state._attempts[change_test.Change(1)])
attempt_count_2 = len(state._attempts[change_test.Change(9)])
self.assertGreaterEqual(attempt_count_1, attempt_count_2)
state.ScheduleWork()
state.Explore()
# We still need more information. Add more attempts to the other Change.
self.assertEqual(len(state._changes), 2)
attempt_count_1 = len(state._attempts[change_test.Change(1)])
attempt_count_2 = len(state._attempts[change_test.Change(9)])
self.assertEqual(attempt_count_1, attempt_count_2)
def testDifferentWithMidpoint(self):
quests = [
quest_test.QuestByChange({
change_test.Change(1):
quest_test.QuestPass(),
change_test.Change(9):
quest_test.QuestFail(),
})
]
state = job_state.JobState(quests,
comparison_mode=job_state.PERFORMANCE)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(9))
state.ScheduleWork()
state.Explore()
# The Changes are different. Add the midpoint.
expected = [
change_test.Change(1),
change_test.Change(5),
change_test.Change(9)
]
self.assertEqual(state._changes, expected)
attempt_count_1 = len(state._attempts[change_test.Change(1)])
attempt_count_2 = len(state._attempts[change_test.Change(5)])
attempt_count_3 = len(state._attempts[change_test.Change(9)])
self.assertEqual(attempt_count_1, attempt_count_2)
self.assertEqual(attempt_count_2, attempt_count_3)
self.assertEqual([], state.Differences())
def testDifferences(self):
quests = [quest_test.QuestPass()]
state = job_state.JobState(
quests, comparison_mode=job_state.FUNCTIONAL, comparison_magnitude=0.2)
change_a = change_test.Change(1)
change_b = change_test.Change(9)
state.AddChange(change_a)
state.AddChange(change_b)
self.assertEqual([], state.Differences())
def testFailedAttempt(self):
quests = [quest_test.QuestPass()]
state = job_state.JobState(quests)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(2))
state.ScheduleWork()
for attempt in state._attempts[change_test.Change(1)]:
attempt._last_execution._exception = "Failed"
self.assertTrue(state.FirstOrLastChangeFailed())
def testNoAttempts(self):
quests = [quest_test.QuestPass()]
state = job_state.JobState(quests)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(2))
state.ScheduleWork()
# It shouldn't happen that a change has no attempts, but we should cope
# gracefully if that somehow happens.
del state._attempts[change_test.Change(1)][:]
del state._attempts[change_test.Change(2)][:]
self.assertFalse(state.FirstOrLastChangeFailed())
def testDifferentNoMidpoint(self):
quests = [quest_test.QuestByChange({
change_test.Change(1): quest_test.QuestPass(),
change_test.Change(2): quest_test.QuestFail(),
})]
state = job_state.JobState(quests, comparison_mode=job_state.PERFORMANCE)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(2))
state.ScheduleWork()
state.Explore()
# The Changes are different, but there's no midpoint. We're done.
self.assertEqual(len(state._changes), 2)
attempt_count_1 = len(state._attempts[change_test.Change(1)])
attempt_count_2 = len(state._attempts[change_test.Change(2)])
self.assertEqual(attempt_count_1, attempt_count_2)
def testSame(self):
quests = [quest_test.QuestPass()]
state = job_state.JobState(quests, comparison_mode=job_state.FUNCTIONAL)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(9))
for _ in xrange(5):
# More Attempts give more confidence that they are, indeed, the same.
state.AddAttempts(change_test.Change(1))
state.AddAttempts(change_test.Change(9))
state.ScheduleWork()
state.Explore()
# The Changes are the same. Do not add any Attempts or Changes.
self.assertEqual(len(state._changes), 2)
attempt_count_1 = len(state._attempts[change_test.Change(1)])
attempt_count_2 = len(state._attempts[change_test.Change(9)])
self.assertEqual(attempt_count_1, attempt_count_2)
def testDifferentWithMidpoint(self):
quests = [
quest_test.QuestByChange({
change_test.Change(1):
quest_test.QuestPass(),
change_test.Change(9):
quest_test.QuestFail(),
})
]
state = job_state.JobState(quests,
comparison_mode=job_state.PERFORMANCE)
state.AddChange(change_test.Change(1))
state.AddChange(change_test.Change(9))
state.ScheduleWork()
state.Explore()
# The Changes are different. Add the speculated modpoints in a two-level
# deep bisection tree, which include 3, 5 (the midpoint between 0 and 9),
# and 7.
expected = [
change_test.Change(1),
change_test.Change(3),
change_test.Change(5),
change_test.Change(7),
change_test.Change(9),
]
# View the whole diff in case of failure.
self.maxDiff = None
self.assertEqual(state._changes, expected)
attempt_count_1 = len(state._attempts[change_test.Change(1)])
attempt_count_2 = len(state._attempts[change_test.Change(3)])
attempt_count_3 = len(state._attempts[change_test.Change(5)])
attempt_count_4 = len(state._attempts[change_test.Change(7)])
attempt_count_5 = len(state._attempts[change_test.Change(9)])
self.assertEqual(attempt_count_1, attempt_count_2)
self.assertEqual(attempt_count_2, attempt_count_3)
self.assertEqual(attempt_count_3, attempt_count_4)
self.assertEqual(attempt_count_4, attempt_count_5)
self.assertEqual([], state.Differences())
def testNoWorkLeft(self): quests = [quest_test.QuestPass()] state = job_state.JobState(quests) state.AddChange(change_test.Change(123)) self.assertTrue(state.ScheduleWork()) self.assertFalse(state.ScheduleWork())
