def test_pop():
"""Test pop method if removes the first value and return it."""
from src.linked_list import LinkedList
linked_list = LinkedList()
linked_list.push('1')
linked_list.push('2')
linked_list.push('3')
popped_item = linked_list.pop()
assert popped_item == '3'
class Stack(object):
"""Stack object for creating a stack list."""
def __init__(self, data=None):
"""Initialize stack class."""
self._container = LinkedList(data)
def push(self, value):
"""Add a value to the stack."""
return self._container.push(value)
def pop(self):
"""Remove a value from the stack and returns that value."""
return self._container.pop()
class Stack(object):
"""Implementation of Stack.
public methods:
push(value) - Adds a value to the stack.
The parameter is the value to be added to the stack.
pop() - Removes a value from the stack and returns that value.
If the stack is empty, attempts to call pop should raise an exception.
"""
def __init__(self, data=None):
"""Initialization."""
self._stack = LinkedList(data)
def push(self, val):
"""Add val to the stack."""
self._stack.push(val)
def pop(self):
"""Remove item off the stack."""
self._stack.pop()
class TestLinkedList(unittest.TestCase):
def setUp(self) -> None:
self.ll = LinkedList()
def test_add_first(self) -> None:
self.ll.add_first(1)
self.assertEqual(self.ll.__str__(), "[1]")
self.ll.add_first(2)
self.assertEqual(self.ll.__str__(), "[2, 1]")
self.ll.add_first(3)
self.assertEqual(self.ll.__str__(), "[3, 2, 1]")
def test_append(self) -> None:
self.ll.append(1)
self.assertEqual(self.ll.__str__(), "[1]")
self.ll.append(2)
self.assertEqual(self.ll.__str__(), "[1, 2]")
self.ll.append(3)
self.assertEqual(self.ll.__str__(), "[1, 2, 3]")
def test_get(self) -> None:
with self.assertRaises(IndexError) as context:
self.ll.get(0)
self.assertTrue("Index is out of range" in context.exception)
self.ll.append(1)
self.assertEqual(self.ll.get(0), 1)
with self.assertRaises(IndexError) as context:
self.ll.get(-1)
self.assertTrue("Index is out of range" in context.exception)
with self.assertRaises(IndexError) as context:
self.ll.get(1)
self.assertTrue("Index is out of range" in context.exception)
self.ll.append(2)
self.assertEqual(self.ll.get(0), 1)
self.assertEqual(self.ll.get(1), 2)
with self.assertRaises(IndexError) as context:
self.ll.get(2)
self.assertTrue("Index is out of range" in context.exception)
self.ll.add_first(3)
self.assertEqual(self.ll.get(0), 3)
self.assertEqual(self.ll.get(1), 1)
self.assertEqual(self.ll.get(2), 2)
with self.assertRaises(IndexError) as context:
self.ll.get(3)
self.assertTrue("Index is out of range" in context.exception)
def test_pop(self) -> None:
with self.assertRaises(IndexError) as context:
self.ll.pop(0)
self.assertTrue("Index is out of range" in context.exception)
self.ll.append(1)
self.assertEqual(self.ll.pop(0), 1)
with self.assertRaises(IndexError) as context:
self.ll.pop(1)
self.assertTrue("Index is out of range" in context.exception)
self.ll.append(1)
self.ll.append(2)
self.assertEqual(self.ll.pop(0), 1)
self.assertEqual(self.ll.pop(0), 2)
self.ll.add_first(3)
self.ll.add_first(2)
self.ll.add_first(1)
self.assertEqual(self.ll.pop(0), 1)
self.assertEqual(self.ll.pop(1), 3)
with self.assertRaises(IndexError) as context:
self.ll.pop(2)
self.assertTrue("Index is out of range" in context.exception)
def test_clear(self) -> None:
self.assertEqual(len(self.ll), 0)
self.ll.append(1)
self.assertEqual(len(self.ll), 1)
self.ll.clear()
self.assertEqual(len(self.ll), 0)
self.ll.append(2)
self.assertEqual(len(self.ll), 1)
self.ll.clear()
self.assertEqual(len(self.ll), 0)
def test_big_test(self) -> None:
test_case = 100_000
for i in range(test_case):
self.ll.add_first(i)
for i in reversed(range(test_case)):
self.assertEqual(self.ll.pop(0), i)
def test_iterable(self) -> None:
for i in range(10):
self.ll.append(i)
for i in self.ll:
self.assertEqual(self.ll[i], i)
self.ll[5] = 6
self.assertEqual(self.ll[5], 6)
def test_repr(self) -> None:
for i in range(6):
self.ll.append(i)
self.assertEqual(self.ll.__repr__(), "LinkedList [0, 1, 2, 3, 4, 5]")
self.ll.append(3)
self.assertEqual(self.ll.__repr__(),
"LinkedList [0, 1, 2, 3, 4, 5, ...]")
