def test_remove(self):
new_stack = Stack()
new_stack.add(5)
new_stack.add(10)
new_stack.add(20)
new_stack.remove()
result = new_stack.return_data()
self.assertEqual(result, [5, 10])
def test_empty(self):
new_stack = Stack()
new_stack.add(60)
result_not_empty = new_stack.is_empty()
self.assertTrue(result_not_empty)
new_stack.remove()
result_empty = new_stack.is_empty()
self.assertFalse(result_empty)
def testSize(self):
s = Stack()
s.add("1")
self.assertTrue(s.size() == 1)
s.remove()
self.assertFalse(s.size() == 1)
def testRemove(self):
s = Stack()
s.add("1")
s.add("2")
s.remove()
self.assertTrue(s.size() == 1)
