Проверить функцию удаления элемента из двунаправленного связанного списка . unittest - том

class Node:
    def __init__(self, v, p=None, n=None):
        self.value = v
        self.prev = p
        self.next = n
 
class LinkedList2:
    def __init__(self):
        self.head = None
        self.tail = None
 
    def add_in_tail(self, item):
        if self.head is None:
            self.head = item
            item.prev = None
            item.next = None
        else:
            self.tail.next = item
            item.prev = self.tail
        self.tail = item
    def delete(self, val, all=False):
        one_run = self.head
        if self.head == None:  # после нахождения 1 го запускать цикл по новой
            return
        if one_run.value == val and self.head.next == None:  # если удаляем один элемент
            self.head = self.tail = None
            return
        if one_run.value == val:  # если удаляем первый элемент
            self.head = self.head.next
            one_run.next.prev = None
            one_run.next = None
            one_run = self.head
            if all == False:
                return
        while one_run is not None:
            if one_run.value == val and one_run.next != None:
                one_run = one_run.prev
                one_run.next = one_run.next.next
                one_run = one_run.next
                one_run.prev = one_run.prev.prev
                one_run = one_run.prev
                if all == False:
                    return
            if one_run.value == val and one_run.next == None:  # если в составе удаленного есть последний элемент
                one_run = one_run.prev
                one_run.next = None  # one_run.next.next
                self.tail = one_run
                if all == False:
                    return
            else:
                one_run = one_run.next

import unittest
 
class MyTestCase(unittest.TestCase):
 
    def test_delete(self):  # сравниваем после удвления элемента в пустой список
        my_list = LinkedList2.LinkedList2()
        my_list.add_in_tail(LinkedList2.Node(54))
        my_list.delete(54)
        self.assertEqual(my_list.head, my_list.tail)
        self.assertEqual(my_list.head, None)
        self.assertEqual(my_list.tail, None)
 
    def test_delete_end(self):  # сравниваем после удвления элемента из середины
        my_list = LinkedList2.LinkedList2()
        x = my_list.add_in_tail(LinkedList2.Node(10))
        my_list.add_in_tail(LinkedList2.Node(20))
        my_list.add_in_tail(LinkedList2.Node(30))
        my_list.add_in_tail(LinkedList2.Node(40))
        my_list.delete(30)
        self.assertEqual(my_list.head, x.self.head , 'хвосты не равны')
        self.assertEqual(my_list.head, None)
        self.assertEqual(my_list.tail, None)
 
if __name__ == '__main__':
    unittest.main()