class test_graph(unittest.TestCase):
def setUp(self):
self.tens = SinglyLinkedList(range(0, 100, 10))
self.blankList = SinglyLinkedList()
def test_length_method(self):
self.assertEqual(len(self.tens), 10)
self.assertEqual(len(self.blankList), 0)
def test_add_method(self):
self.blankList.append(50)
self.tens.append(110)
self.assertEqual(len(self.blankList), 1)
self.assertEqual(len(self.tens), 11)
class TestSinglyLinkedList(TestCase):
"""
Test cases to test the collection wrapper
"""
def setUp(self):
self.sll = SinglyLinkedList()
self.reverse = getattr(self.sll, 'reverse_iterative')
def test_empty_list(self):
self.assertEqual(len(self.sll), 0)
def test_list_single_item(self):
self.sll.append(1)
self.assertEqual(len(self.sll), 1)
for v in self.sll:
self.assertEqual(v, 1)
def test_list_multi_items(self):
num = 10
for i in range(num):
self.sll.append(i)
for i, list_val in zip(range(num), self.sll):
self.assertEqual(i, list_val)
def test_to_str(self):
self.sll.append(1)
self.sll.append(2)
self.sll.append(3)
self.assertEqual(str(self.sll), str([1, 2, 3]))
def test_to_repr(self):
self.sll.append(1)
self.sll.append(2)
self.sll.append(3)
self.assertEqual(repr(self.sll), "{}({})".format("SinglyLinkedList",
repr([1, 2, 3])))
