def test_optimize_1(self):
campaign = Campaign(2, 100, 3)
campaign.set_sub_campaign_values(0, [3, 7, 14])
campaign.set_sub_campaign_values(1, [2, 5, 7])
opt_matrix, max_idx_matrix = CampaignOptimizer._optimize(campaign)
true_opt_matrix = np.array([[0, 0, 0],
[3, 7, 14],
[5, 9, 16]])
true_max_idx_matrix = np.array([[-1, -1, -1],
[0, 1, 2],
[0, 0, 0]])
self.assertTrue((true_opt_matrix == opt_matrix).all())
self.assertTrue((true_max_idx_matrix == max_idx_matrix).all())
def test_optimize_2(self):
campaign = Campaign(3, 90, 4)
campaign.set_sub_campaign_values(0, [0, 3, 12, 20])
campaign.set_sub_campaign_values(1, [0, 2, 7, 10])
campaign.set_sub_campaign_values(2, [0, 5, 8, 12])
opt_matrix, max_idx_matrix = CampaignOptimizer._optimize(campaign)
true_opt_matrix = np.array([[0, 0, 0, 0],
[0, 3, 12, 20],
[0, 3, 12, 20],
[0, 5, 12, 20]])
true_max_idx_matrix = np.array([[-1, -1, -1, -1],
[0, 1, 2, 3],
[0, 0, 0, 0],
[0, 1, 0, 0]])
self.assertTrue((true_opt_matrix == opt_matrix).all())
self.assertTrue((true_max_idx_matrix == max_idx_matrix).all())
def test_optimize_3(self):
campaign = Campaign(4, 100, 5)
campaign.set_sub_campaign_values(0, [0, 3, 12, 20, 15])
campaign.set_sub_campaign_values(1, [0, 2, 7, 10, 9])
campaign.set_sub_campaign_values(2, [0, 5, 8, 12, 18])
campaign.set_sub_campaign_values(3, [0, 9, 9, 10, 7])
opt_matrix, max_idx_matrix = CampaignOptimizer._optimize(campaign)
true_opt_matrix = np.array([[0, 0, 0, 0, 0],
[0, 3, 12, 20, 20],
[0, 3, 12, 20, 22],
[0, 5, 12, 20, 25],
[0, 9, 14, 21, 29]])
true_max_idx_matrix = np.array([[-1, -1, -1, -1, -1],
[0, 1, 2, 3, 3],
[0, 0, 0, 0, 1],
[0, 1, 0, 0, 1],
[0, 1, 1, 1, 1]])
self.assertTrue((true_opt_matrix == opt_matrix).all())
self.assertTrue((true_max_idx_matrix == max_idx_matrix).all())