class TestStack(unittest.TestCase):
def setUp(self):
self.n1 = Node(1)
self.n2 = Node(2)
self.n3 = Node(3)
self.stack = LinkedListStack()
self.stack.push(1)
self.stack.push(2)
self.stack.push(self.n1)
def test_peek(self):
item = self.stack.peek()
self.assertEqual(item, self.n1)
def test_push(self):
self.assertEqual(self.stack.get_size(), 3)
self.stack.push(self.n2)
self.assertEqual(self.stack.get_size(), 4)
def test_pop(self):
item = self.stack.pop()
self.assertEqual(item, self.n1)
def test_is_empty(self):
empty_stack = LinkedListStack()
self.assertEqual(empty_stack.is_empty(), True)
self.assertEqual(self.stack.is_empty(), False)
def test_get_size(self):
empty_stack = LinkedListStack()
self.assertEqual(empty_stack.get_size(), 0)
self.assertEqual(self.stack.get_size(), 3)
def test_get_size(self):
empty_stack = LinkedListStack()
self.assertEqual(empty_stack.get_size(), 0)
self.assertEqual(self.stack.get_size(), 3)