def test_size(self): stack = Stack() self.assertEqual(stack.size(), 0) stack.push(1) self.assertEqual(stack.size(), 1) stack.push(2) self.assertEqual(stack.size(), 2)
def test_push(self): stack = Stack() stack.push(5) self.assertEqual(stack.data, [5]) self.assertEqual(stack.size(), 1) stack.push(5) self.assertEqual(stack.data, [5, 5]) self.assertEqual(stack.size(), 2)
def test_peek(self): stack = Stack() self.assertRaises(Exception, stack.peek) stack.push(5) stack.push(4) self.assertEqual(stack.peek(), 4) self.assertEqual(stack.size(), 2) stack.pop() self.assertEqual(stack.peek(), 5) self.assertEqual(stack.size(), 1)