def test_remove(self):
lst = ArrayList()
self.assertEqual(remove(lst, 0), ArrayList())
lst.arr = [1, 2, None, None]
lst.capacity = 4
lst.num_items = 2
self.assertEqual(remove(lst, 2).arr, [1, None])
lst.arr = [1, 2, 3, None]
lst.capacity = 4
lst.num_items = 3
self.assertEqual(remove(lst, 2).arr, [1, 3, None, None])
def test_array_list6(self):
lst = ArrayList()
for i in range(3):
lst = array_list.insert(lst, i, i)
lst2 = ArrayList()
lst2.arr = [0, 1, 2, None]
lst2.capacity = 4
lst2.num_items = 3
self.assertEqual(lst, lst2)
lst = array_list.remove(lst, 1)
self.assertEqual(lst.capacity, lst2.capacity)
lst2.arr = [2, None]
lst2.capacity = 2
lst2.num_items = 1
self.assertEqual(array_list.remove(lst, 0), lst2)
def test_pop(self):
lst = ArrayList()
lst.num_items = 2
lst.arr = [1, 2]
lst2 = ArrayList()
lst2.num_items = 1
lst2.arr = [1, None]
self.assertEqual(pop(lst, 1), (lst2, 2))
self.assertRaises(IndexError, pop, lst, 2)
lst.arr = [1, 2, None, None]
lst.capacity = 4
lst.num_items = 2
lst2.arr = [2, None]
lst2.num_items = 1
lst2.capacity = 2
self.assertEqual(pop(lst, 0)[0].arr, [2, None])
def test_insert(self):
lst = ArrayList()
self.assertEqual(insert(lst, 5, 0).arr, [5, None])
lst.num_items = 4
lst.capacity = 4
lst.arr = [1, 2, 3, 4]
self.assertEqual(insert(lst, 5, 2).arr, [1, 2, 5, 3, 4, None, None, None])
self.assertRaises(IndexError, insert, lst, 20, 6)
def test_array_list6(self):
lst = ArrayList()
for i in range(3):
lst = array_list.insert(lst, i, i)
lst2 = ArrayList()
lst2.arr = [0, 1, 2, None]
lst2.capacity = 4
lst2.num_items = 3
self.assertEqual(lst, lst2)
lst = array_list.remove(lst, 1)
self.assertEqual(lst.capacity, lst2.capacity)
lst2.arr = [2, None]
lst2.capacity = 2
lst2.num_items = 1
print("L1: ", lst.arr)
print("L2: ", lst2.arr)
# self.assertEqual(array_list.remove(lst, 0), lst2)
array_list.remove(lst, 0)
print("Removed 0 from list 1")
print("L1: ", lst.arr)
print("L2: ", lst2.arr)
def test_shrink(self):
lst1 = ArrayList()
lst1.capacity = 6
lst1 = shrink(lst1)
self.assertEqual(lst1.capacity, 3)
def test_enlarge(self):
lst1 = enlarge(ArrayList())
lst2 = ArrayList()
lst2.capacity = 4
self.assertEqual(lst1.capacity, lst2.capacity)
