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])))