def setUp(self):
test_helpers.patch_environ(self, env={'APP_ARGS': '--dummy'})
data_types.Trial(app_name='app_1', probability=0.5,
app_args='--a1').put()
data_types.Trial(app_name='app_2', probability=0.4,
app_args='--a2').put()
data_types.Trial(app_name='app_2', probability=0.2,
app_args='--b2').put()
data_types.Trial(app_name='app_3', probability=1.0,
app_args='--a3').put()
test_helpers.patch(self, ['bot.tasks.trials.select_trial'])
def setUp(self):
test_helpers.patch_environ(self, env={'APP_ARGS': '-x'})
data_types.Trial(app_name='app_1', probability=0.5,
app_args='--a1').put()
data_types.Trial(app_name='app_2', probability=0.4,
app_args='--b1').put()
data_types.Trial(app_name='app_2', probability=0.2,
app_args='--b2').put()
data_types.Trial(app_name='app_3', probability=1.0,
app_args='--c1').put()
data_types.Trial(app_name='app_3', probability=0.2,
app_args='--c2').put()
data_types.Trial(app_name='app_3', probability=0.2,
app_args='--c3').put()
test_helpers.patch(self, ['random.random'])
def test_select_trial_single_item(self):
"""Ensure that we return a Trial if we select one."""
self.mock.random.return_value = 0.0
trial = data_types.Trial(probability=0.5)
self.assertEqual(trials.select_trial([trial]), trial)
def test_select_trial_multiple_items(self):
"""Ensure that we can select a Trial from a list of multiple."""
self.mock.random.return_value = 0.5
trial_1 = data_types.Trial(probability=0.5)
trial_2 = data_types.Trial(probability=0.3)
self.assertEqual(trials.select_trial([trial_1, trial_2]), trial_2)
def test_select_trial_not_selected(self):
"""Ensure that we return None if we did not randomly select a trial."""
self.mock.random.return_value = 0.5
trial = data_types.Trial(probability=0.5)
self.assertIsNone(trials.select_trial([trial]))