Beispiel #1
0
class LinkedListTests(unittest.TestCase):
    def setUp(self):
        self.list = LinkedList()

    def test_add_to_tail(self):
        self.list.add_to_tail(1)
        self.assertEqual(self.list.tail.value, 1)
        self.assertEqual(self.list.head.value, 1)
        self.list.add_to_tail(2)
        self.assertEqual(self.list.tail.value, 2)
        self.assertEqual(self.list.head.value, 1)

    def test_contains(self):
        self.list.add_to_tail(1)
        self.list.add_to_tail(2)
        self.list.add_to_tail(5)
        self.list.add_to_tail(10)
        self.assertTrue(self.list.contains(10))
        self.assertTrue(self.list.contains(2))
        self.assertFalse(self.list.contains(1000))

    def test_remove_head(self):
        self.list.add_to_tail(10)
        self.list.add_to_tail(20)
        self.assertEqual(self.list.remove_head(), 10)
        self.assertFalse(self.list.contains(10))
        self.assertEqual(self.list.remove_head(), 20)
        self.assertFalse(self.list.contains(20))

        self.list.add_to_tail(10)
        self.assertEqual(self.list.remove_head(), 10)
        self.assertIsNone(self.list.head)
        self.assertIsNone(self.list.tail)
        self.assertIsNone(self.list.remove_head())

    def test_get_max(self):
        self.assertIsNone(self.list.get_max())
        self.list.add_to_tail(100)
        self.assertEqual(self.list.get_max(), 100)
        self.list.add_to_tail(55)
        self.assertEqual(self.list.get_max(), 100)
        self.list.add_to_tail(101)
        self.assertEqual(self.list.get_max(), 101)

    def test_remove_tail(self):
        print(self.list)
        self.list.add_to_tail(1)
        self.list.add_to_tail(2)
        self.list.add_to_tail(5)
        self.list.add_to_tail(10)
        print(self.list)
        self.assertEqual(self.list.remove_tail(), 10)
        print(self.list)
Beispiel #2
0
class LinkedListTests(unittest.TestCase):
    def setUp(self):
        self.list = LinkedList()

    def test_add_to_tail(self):
        self.list.add_to_tail(1)
        self.assertEqual(self.list.tail.value, 1)
        self.assertEqual(self.list.head.value, 1)
        self.list.add_to_tail(2)
        self.assertEqual(self.list.tail.value, 2)
        self.assertEqual(self.list.head.value, 1)

    def test_remove_head(self):
        self.list.add_to_tail(10)
        self.list.add_to_tail(20)
        self.assertEqual(self.list.remove_head(), 10)
        self.assertEqual(self.list.remove_head(), 20)

        self.list.add_to_tail(10)
        self.assertEqual(self.list.remove_head(), 10)
        self.assertIsNone(self.list.head)
        self.assertIsNone(self.list.tail)
        self.assertIsNone(self.list.remove_head())

    def test_remove_tail(self):
        self.list.add_to_tail(30)
        self.list.add_to_tail(40)
        self.assertEqual(self.list.remove_tail(), 40)
        self.assertEqual(self.list.remove_tail(), 30)

        self.list.add_to_tail(100)
        self.assertEqual(self.list.remove_tail(), 100)
        self.assertIsNone(self.list.head)
        self.assertIsNone(self.list.tail)
        self.assertIsNone(self.list.remove_tail())

    def test_get_max(self):
        self.list.add_to_tail(30)
        self.list.add_to_tail(40)
        self.assertEqual(self.list.get_max(), 40)
        self.assertEqual(self.list.remove_tail(), 40)
        self.assertEqual(self.list.get_max(), 30)
        self.assertEqual(self.list.remove_tail(), 30)
        self.assertEqual(self.list.get_max(), None)