def test_two_empty_heaps(self):
inNum = 4
hLo = []
hHi = []
resLo, resHi = med_maint.insert_num(inNum, hLo, hHi)[:2]
self.assertEqual(resLo, hLo)
self.assertEqual(resHi, [4])
def test_misc(self):
inNum = 3
hLo = [-1]
hHi = [2]
resMed = med_maint.insert_num(inNum, hLo, hHi)[-1]
ansMed = 2
self.assertEqual(resMed, ansMed)
def test_single_and_empty_insert_to_hi(self):
inNum = 4
hLo = []
hHi = [3]
resMed = med_maint.insert_num(inNum, hLo, hHi, equalLen=False)[-1]
ansMed = 3
self.assertEqual(resMed, ansMed)
def test_single_and_empty_insert_to_lo(self):
inNum = 4
hLo = []
hHi = [7]
resLo, resHi = med_maint.insert_num(inNum, hLo, hHi, equalLen=False)[:2]
ansLo = [-4]
self.assertEqual(resLo, ansLo)
self.assertEqual(resHi, hHi)
def test_two_unequal_len_insert_to_hi(self):
inNum = 4
hLo = [-1]
hHi = [2, 3]
heapq.heapify(hHi)
resMed = med_maint.insert_num(inNum, hLo, hHi, equalLen=False)[-1]
ansMed = 2
self.assertEqual(resMed, ansMed)
def test_two_equal_len_insert_to_hi(self):
inNum = 5
hLo = [-1, -2]
heapq.heapify(hLo)
hHi = [3, 4]
heapq.heapify(hHi)
resMed = med_maint.insert_num(inNum, hLo, hHi)[-1]
ansMed = 3
self.assertEqual(resMed, ansMed)
def test_two_unequal_len_insert_to_lo(self):
inNum = 2
hLo = [-1]
hHi = [3, 4]
heapq.heapify(hHi)
resLo, resHi = med_maint.insert_num(inNum, hLo, hHi, equalLen=False)[:2]
ansLo = [-1, -2]
heapq.heapify(ansLo)
self.assertEqual(resLo, ansLo)
self.assertEqual(resHi, hHi)
def test_two_equal_len_insert_to_hi(self):
inNum = 5
hLo = [-1, -2]
heapq.heapify(hLo)
hHi = [3, 4]
heapq.heapify(hHi)
resLo, resHi = med_maint.insert_num(inNum, hLo, hHi)[:2]
ansHi = [3, 4, 5]
heapq.heapify(ansHi)
self.assertEqual(resLo, hLo)
while ansHi:
self.assertEqual(heapq.heappop(resHi), heapq.heappop(ansHi))
def test_two_empty_heaps(self):
inNum = 4
hLo = []
hHi = []
resMed = med_maint.insert_num(inNum, hLo, hHi)[-1]
self.assertEqual(resMed, 4)
