def test_remove_when_single_element_in_myarray(self): # Arrange myarray = MyArray('i', 20) # Act myarray.remove(20) # Assert self.assertEqual(0, myarray.size())
def test_remove_when_more_than_one_element_in_myarray(self): # Arrange myarray = MyArray('i', 20, 30) # Act myarray.remove(20) # Assert self.assertEqual(1, myarray.size()) self.assertEqual(30, myarray.get(0))