class ElfTest(unittest.TestCase):
def setUp(self):
self.elf1 = Elf(1)
self.elf2 = Elf(2)
# soln_file = os.path.join(os.getcwd(), 'test.csv')
# self.wcsv = csv.writer(open(soln_file, "wb"))
def test_num_of_working_minutes_left(self):
self.assertEquals(self.elf1.num_of_working_minutes_left(), 600)
self.elf1.set_next_available_time(600)
self.assertEquals(self.elf1.num_of_working_minutes_left(), 600-60)
self.elf1.set_next_available_time(540 + 595)
self.assertEquals(self.elf1.num_of_working_minutes_left(), 5)
self.elf1.set_next_available_time(540 + 605)
self.assertEquals(self.elf1.num_of_working_minutes_left(), 0)
def test_get_productivity(self):
self.assertEquals(self.elf1.get_productivity(), 1.0)
self.elf1.set_productivity(2.0)
self.assertEquals(self.elf1.get_productivity(), 2.0)
def test_wait_till_next_day(self):
self.assertEquals(self.elf1.get_next_available_time(), 540)
self.elf1.set_next_available_time(700)
self.assertEquals(self.elf1.get_next_available_time(), 700)
self.elf1.wait_till_next_day()
self.assertEquals(self.elf1.get_next_available_time(), 1440+(9*60))
self.elf1.wait_till_next_day()
self.assertEquals(self.elf1.get_next_available_time(), (2*1440)+(9*60))
# Apply strategy for all elves in pool
# Pour chaque elfe:
# 1 jouet au hasard parmi ceux disponibles
# Si on peut traiter l'objet dans la journée, le faire et mettre à jour la date de disponibilité de l'elfe
# Sinon planifier l'objet pour le lendemain matin et le traiter et remplir le reste de la journée avec des objets "courts"
if c % 10000 == 0:
print("TOYPOOL WAITING LEN : %d, TOYPOOL AVAILABLE LEN : %d" % (mytoypool.length_waiting_list(), mytoypool.length_available_list()))
# Etape 1 Bis : Mettre à jour le toy pool par rapport à la date de disponibilité de l'elfe
# Avancer jusqu'à ce qu'il y ait des jouets dans l'available list
while True:
mytoypool.update_available_toy_list_according_to_elf(elf)
if mytoypool.length_available_list() > 0:
break
else:
t = elf.get_next_available_time()
next_t = hrs.next_sanctioned_minute(t)
elf.set_next_available_time(next_t)
# Etape 2 : Recupérer la productivité de l'elfe
productivity = elf.get_productivity()
# Gros jouet au hasard
r = random.random()
if r > RATIO_RANDOM:
toy = mytoypool.get_next_longest_toy_for_elf(elf)
elf.make_toy(toy, wcsv)
continue
# Etape 2, Cas 1 : La productivité est supérieure au seuil