Проблема с тестом (unittest)

Не работает тест:

import unittest
from main import search_name, add_document, del_document
class TestFooDocumentsDirectories(unittest.TestCase):
    def setUp(self):
        print("method setup")

    def tearDown(self):
        print("method setup")

    def test_search_name(self):
        documents = [
            {"type": "passport", "number": "2207 876234", "name": "Василий Гупкин"},
            {"type": "invoice", "number": "11-2", "name": "Геннадий Покемонов"},
            {"type": "insurance", "number": "10006", "name": "Аристарх Павлов"}
        ]
        self.assertEqual(search_name(documents, "11-2"), "Геннадий Покемонов")

    def test_add_document(self):
        documents = [
            {"type": "passport", "number": "2207 876234", "name": "Василий Гупкин"},
            {"type": "invoice", "number": "11-2", "name": "Геннадий Покемонов"},
            {"type": "insurance", "number": "10006", "name": "Аристарх Павлов"}
        ]
        directories = {
            '1': ['2207 876234', '11-2'],
            '2': ['10006'],
            '3': []
        }
        self.assertIn(add_document(documents, directories, "1", "43", "Военный билет", "Геннадий Гудков"), 43)

    if __name__ == '__main__':
        unittest.main()

со следующей функцией:

    def add_document(list_documents, list_directories, number_shelf, number, type, name):
      for key,value in list_directories.items():
        if number_shelf==key:
          date_document=dict(type=type, number=number, name=name)
          list_documents.append(date_document)
          value.append(number)
          return list_documents, list_directories
      else:
          return("Вы ввели номер несуществующей полки")

функция добавляет в список documents словарь с данными по документу в следующем формате: {"type": "insurance", "number": "10006", "name": "Аристарх Павлов"} при условии наличия переданного в параметре функции номера полки в словаре directories.

    documents = [
            {"type": "passport", "number": "2207 876234", "name": "Василий Гупкин"},
            {"type": "invoice", "number": "11-2", "name": "Геннадий Покемонов"},
            {"type": "insurance", "number": "10006", "name": "Аристарх Павлов"}
        ]
    directories = {
            '1': ['2207 876234', '11-2'],
            '2': ['10006'],
            '3': []
        }