def test_duration_update(self):
elf = Elf(1)
elf_future = Elf(2)
elf_future.wait_till_next_day()
toypool = ToyPool()
toy1 = Toy(1, "2014 1 1 8 0", 116)
toy2 = Toy(2, "2014 1 1 8 0", 116)
toy3 = Toy(3, "2014 1 1 8 0", 117)
toy4 = Toy(4, "2014 1 1 8 0", 117)
toy5 = Toy(5, "2014 1 1 8 0", 118)
toy6 = Toy(6, "2014 1 1 8 0", 118)
toypool.push_toy_in_waiting_list(toy1)
toypool.push_toy_in_waiting_list(toy2)
toypool.push_toy_in_waiting_list(toy3)
toypool.push_toy_in_waiting_list(toy4)
toypool.push_toy_in_waiting_list(toy5)
toypool.push_toy_in_waiting_list(toy6)
toypool.update_available_toy_list_according_to_elf(elf)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
toy = toypool.get_next_shortest_toy_for_elf(elf)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
toy = toypool.get_next_longest_toy_for_elf(elf)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
toy6 = Toy(6, "2014 1 1 9 5", 118)
toypool.push_toy_in_waiting_list(toy6)
toypool.update_available_toy_list_according_to_elf(elf_future)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
#print toypool.get_available_toy_duration()
#print toypool.get_hash_toy_duration_timestamp()
toy = toypool.get_toy_by_duration_for_elf(elf, 118)
toy = toypool.get_toy_by_duration_for_elf(elf, 118)
toy = toypool.get_toy_by_duration_for_elf(elf, 118)
#print toypool.get_available_toy_duration()
#print toypool.get_hash_toy_duration_timestamp()
self.assertEquals(toypool.get_available_toy_duration(), [116, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 118])
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))
class ToyPoolTest(unittest.TestCase):
def setUp(self):
toy_file = os.path.join(os.getcwd(), '..', 'DATA', 'toys_rev2.csv')
self.toy_empty_pool = ToyPool()
self.toy_filled_pool = ToyPool()
self.toy_filled_pool.add_file_content(toy_file, 10)
self.elf = Elf(1)
self.toy_small_pool = ToyPool()
self.toy1 = Toy(1, "2014 1 1 8 0", 600)
self.toy2 = Toy(2, "2014 1 1 9 3", 60)
self.toy3 = Toy(3, "2014 1 1 10 0", 2)
self.toy4 = Toy(4, "2014 1 1 9 5", 60)
self.toy_small_pool.push_toy_in_waiting_list(self.toy1)
self.toy_small_pool.push_toy_in_waiting_list(self.toy2)
self.toy_small_pool.push_toy_in_waiting_list(self.toy3)
self.toy_small_pool.push_toy_in_waiting_list(self.toy4)
def test_walk_all_durations(self):
toy_pool = ToyPool()
toy1 = Toy(1, "2014 1 1 9 5", 30)
toy2 = Toy(2, "2014 1 1 10 5", 20)
toy3 = Toy(3, "2014 1 1 8 5", 10)
toy_pool.push_toy_in_waiting_list(toy1)
toy_pool.push_toy_in_waiting_list(toy2)
toy_pool.push_toy_in_waiting_list(toy3)
elf_pool = ElfPool(2)
elf1 = elf_pool.next_available_elf()
toy_pool.update_available_toy_list_according_to_elf(elf1) # 1 et 2 dans available
# On prends un 31
toy = toy_pool.get_toy_by_duration_for_elf(elf1, 30)
self.assertEquals(toy, toy3)
def test_behaviour(self):
toy_pool = ToyPool()
toy1 = Toy(1, "2014 1 1 9 5", 20)
toy2 = Toy(2, "2014 1 1 8 5", 30)
toy3 = Toy(3, "2014 1 1 8 3", 15)
toy_pool.push_toy_in_waiting_list(toy1)
toy_pool.push_toy_in_waiting_list(toy2)
toy_pool.push_toy_in_waiting_list(toy3)
elf_pool = ElfPool(2)
elf1 = elf_pool.next_available_elf()
toy_pool.update_available_toy_list_according_to_elf(elf1) # 1 et 2 dans available
# on essaye de prendre un duration 10
toy = toy_pool.get_toy_by_duration_for_elf(elf1, 10)
self.assertTrue(toy is None)
# on essaye de prendre un duration 30
toy = toy_pool.get_toy_by_duration_for_elf(elf1,30)
self.assertEquals(toy, toy2)
# on pousse l'elfe au lendemain
elf1.wait_till_next_day()
# on mets à jour
toy_pool.update_available_toy_list_according_to_elf(elf1)
# changment d'elfe
elf2 = elf_pool.next_available_elf()
# On essaye de prendre un duration 14
toy = toy_pool.get_toy_by_duration_for_elf(elf2, 14)
self.assertTrue(toy is None)
# On essaye de prendre un duration 21
toy = toy_pool.get_toy_by_duration_for_elf(elf2, 21)
self.assertEquals(toy, toy3)
def test_duration_update(self):
elf = Elf(1)
elf_future = Elf(2)
elf_future.wait_till_next_day()
toypool = ToyPool()
toy1 = Toy(1, "2014 1 1 8 0", 116)
toy2 = Toy(2, "2014 1 1 8 0", 116)
toy3 = Toy(3, "2014 1 1 8 0", 117)
toy4 = Toy(4, "2014 1 1 8 0", 117)
toy5 = Toy(5, "2014 1 1 8 0", 118)
toy6 = Toy(6, "2014 1 1 8 0", 118)
toypool.push_toy_in_waiting_list(toy1)
toypool.push_toy_in_waiting_list(toy2)
toypool.push_toy_in_waiting_list(toy3)
toypool.push_toy_in_waiting_list(toy4)
toypool.push_toy_in_waiting_list(toy5)
toypool.push_toy_in_waiting_list(toy6)
toypool.update_available_toy_list_according_to_elf(elf)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
toy = toypool.get_next_shortest_toy_for_elf(elf)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
toy = toypool.get_next_longest_toy_for_elf(elf)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
toy6 = Toy(6, "2014 1 1 9 5", 118)
toypool.push_toy_in_waiting_list(toy6)
toypool.update_available_toy_list_according_to_elf(elf_future)
self.assertEquals(toypool.get_available_toy_duration(), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 117, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 117, 118])
#print toypool.get_available_toy_duration()
#print toypool.get_hash_toy_duration_timestamp()
toy = toypool.get_toy_by_duration_for_elf(elf, 118)
toy = toypool.get_toy_by_duration_for_elf(elf, 118)
toy = toypool.get_toy_by_duration_for_elf(elf, 118)
#print toypool.get_available_toy_duration()
#print toypool.get_hash_toy_duration_timestamp()
self.assertEquals(toypool.get_available_toy_duration(), [116, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_timestamp().keys()), [116, 118])
self.assertEquals(sorted(toypool.get_hash_toy_duration_values().keys()), [116, 118])
#print toypool.get_hash_toy_duration_timestamp()
def test_get_next_longest_toy(self):
self.elf.wait_till_next_day()
self.toy_small_pool.update_available_toy_list_according_to_elf(self.elf)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 0)
self.assertEquals(self.toy_small_pool.length_available_list(), 4)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2, 60, 600])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_timestamp().keys()), [2, 60, 600])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_values().keys()), [2, 60, 600])
toy = self.toy_small_pool.get_next_longest_toy_for_elf(self.elf)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2, 60])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_timestamp().keys()), [2, 60])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_values().keys()), [2, 60])
self.assertEquals(toy, self.toy1)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 0)
self.assertEquals(self.toy_small_pool.length_available_list(), 3)
toy = self.toy_small_pool.get_next_longest_toy_for_elf(self.elf)
self.assertEquals(toy.get_duration(), 60)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2, 60])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_timestamp().keys()), [2, 60])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_values().keys()), [2, 60])
toy = self.toy_small_pool.get_next_longest_toy_for_elf(self.elf)
self.assertEquals(toy.get_duration(), 60)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_timestamp().keys()), [2])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_values().keys()), [2])
toy = self.toy_small_pool.get_next_longest_toy_for_elf(self.elf)
self.assertEquals(toy.get_duration(), 2)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_timestamp().keys()), [])
self.assertEquals(sorted(self.toy_small_pool.get_hash_toy_duration_values().keys()), [])
toy = self.toy_small_pool.get_next_longest_toy_for_elf(self.elf)
self.assertTrue(toy is None)
def test_get_next_shortest_toy(self):
self.elf.wait_till_next_day()
self.toy_small_pool.update_available_toy_list_according_to_elf(self.elf)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 0)
self.assertEquals(self.toy_small_pool.length_available_list(), 4)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2, 60, 600])
toy = self.toy_small_pool.get_next_shortest_toy_for_elf(self.elf)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [60, 600])
self.assertEquals(toy, self.toy3)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 0)
self.assertEquals(self.toy_small_pool.length_available_list(), 3)
toy = self.toy_small_pool.get_next_shortest_toy_for_elf(self.elf)
self.assertEquals(toy.get_duration(), 60)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [60, 600])
toy = self.toy_small_pool.get_next_shortest_toy_for_elf(self.elf)
self.assertEquals(toy.get_duration(), 60)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [600])
toy = self.toy_small_pool.get_next_shortest_toy_for_elf(self.elf)
self.assertEquals(toy.get_duration(), 600)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [])
toy = self.toy_small_pool.get_next_shortest_toy_for_elf(self.elf)
self.assertTrue(toy is None)
def test_push_toy_in_available_list(self):
self.assertEquals(self.toy_small_pool.length_waiting_list(), 4)
self.assertEquals(self.toy_small_pool.length_available_list(), 0)
toy_timestamp, toy = self.toy_small_pool.pop_toy_from_waiting_list()
self.toy_small_pool.push_toy_in_available_list(toy)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 3)
self.assertEquals(self.toy_small_pool.length_available_list(), 1)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [600])
self.assertTrue(self.toy_small_pool.has_available_toy(self.toy1))
self.assertFalse(self.toy_small_pool.has_available_toy(self.toy2))
self.assertFalse(self.toy_small_pool.has_available_toy(self.toy3))
self.assertFalse(self.toy_small_pool.has_available_toy(self.toy4))
self.elf.wait_till_next_day()
self.toy_small_pool.update_available_toy_list_according_to_elf(self.elf)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 0)
self.assertEquals(self.toy_small_pool.length_available_list(), 4)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2, 60, 600])
self.assertTrue(self.toy_small_pool.has_available_toy(self.toy1))
self.assertTrue(self.toy_small_pool.has_available_toy(self.toy2))
self.assertTrue(self.toy_small_pool.has_available_toy(self.toy3))
self.assertTrue(self.toy_small_pool.has_available_toy(self.toy4))
toy_of_duration_1 = self.toy_small_pool.get_toy_by_duration_for_elf(self.elf, 1)
self.assertTrue(toy_of_duration_1 is None)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [2, 60, 600])
toy_of_duration_3 = self.toy_small_pool.get_toy_by_duration_for_elf(self.elf, 3)
self.assertEquals(toy_of_duration_3, self.toy3)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [60, 600])
self.assertEquals(self.toy_small_pool.length_available_list(), 3)
toy_of_duration_50 = self.toy_small_pool.get_toy_by_duration_for_elf(self.elf, 50)
self.assertTrue(toy_of_duration_50 is None)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [60, 600])
toy_of_duration_60 = self.toy_small_pool.get_toy_by_duration_for_elf(self.elf, 60)
self.assertTrue(toy_of_duration_60 is not None)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [60, 600])
toy_of_duration_65 = self.toy_small_pool.get_toy_by_duration_for_elf(self.elf, 65)
self.assertTrue(toy_of_duration_65 is not None)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [600])
# Jouet trop grand
toy_of_duration_650 = self.toy_small_pool.get_toy_by_duration_for_elf(self.elf, 650)
self.assertTrue(toy_of_duration_650 is not None)
self.assertEquals(self.toy_small_pool.get_available_toy_duration(), [])
def test_pop_toy_from_waiting_list(self):
self.assertEquals(self.toy_small_pool.length_waiting_list(), 4)
toy_timestamp, toy = self.toy_small_pool.pop_toy_from_waiting_list()
self.assertEquals(toy_timestamp, 8*60)
self.assertEquals(toy, self.toy1)
self.assertEquals(self.toy_small_pool.length_waiting_list(), 3)
def test_empty(self):
self.assertTrue(self.toy_empty_pool.empty())
elf.set_next_available_time(next_t)
# Etape 2, Cas 2 : La productivité est inférieure au seuil
else:
# Etape 2.2 : Calculer le nombre de minutes restantes avant la fin de la journée
minutes_left = elf.num_of_working_minutes_left()
# Etape 3.2 : Avec la productivité, calculer le nombre de minutes max pouvant être consacrées à un jouet
productivity = elf.get_productivity()
toy_max_minutes = int(minutes_left*productivity)
# Ca de rejet
if toy_max_minutes <= 1:
# On remet ça à demain
elf.wait_till_next_day()
continue
# Etape 4.2 : Selectionner un jouet disposant d'un nombre de minutes inférieur à toy_max_minutes
toy_expected_duration = random.randint(1, toy_max_minutes-1)
toy_expected_duration = toy_max_minutes - 1
toy = mytoypool.get_toy_by_duration_for_elf(elf, toy_expected_duration)
# toy = mytoypool.get_next_shortest_toy_for_elf(elf)
# Etape 4.2, Cas 1 : Un jouet de ce type n'existe pas
if toy is None:
# On est à la fin de la journée ?
if minutes_left < MINUTES_LEFT_END_OF_DAY:
# Mieux vaut attendre demain
elf.wait_till_next_day()