def RunSetStories(self):
possible_stories = [
s for s in system_health_stories.IterAllSystemHealthStoryClasses()
if (s.NAME not in self._BLACKLIST and not s.ABSTRACT_STORY
and self._PLATFORM in s.SUPPORTED_PLATFORMS)
]
assert (self._num_training + self._num_variations * self._num_testing
<= len(possible_stories)), \
'We only have {} stories to work with, but want {} + {}*{}'.format(
len(possible_stories), self._num_training, self._num_variations,
self._num_testing)
if self._run_set == self.DEBUGGING:
return random.sample(possible_stories, 3)
random.shuffle(possible_stories)
if self._run_set == self.TRAINING:
return possible_stories[:self._num_training]
elif self._run_set == self.TESTING:
return possible_stories[(
self._num_training +
self._test_variation * self._num_testing):(
self._num_training +
(self._test_variation + 1) * self._num_testing)]
assert False, 'Bad run set {}'.format(self._run_set)
def __init__(self, platform, take_memory_measurement=True):
super(OrderfileMemory.OrderfileMemoryStorySet,
self).__init__(archive_data_file=(
'../../page_sets/data/system_health_%s.json' % platform),
cloud_storage_bucket=story.PARTNER_BUCKET)
assert platform in platforms.ALL_PLATFORMS
for story_class in system_health_stories.IterAllSystemHealthStoryClasses(
):
if (story_class.ABSTRACT_STORY
or platform not in story_class.SUPPORTED_PLATFORMS
or story_class.NAME not in self._STORY_SET):
continue
self.AddStory(story_class(self, take_memory_measurement))