Как правильно импортировать модули в данном случае?

Question

[Ranc58]

Имеется структура:

project/
    prj_dir/
        main.py
        work_dir/
            __init__.py
            foo.py
            bar.py
     tests.py

Импорт в .py файлах:

#main.py 
from work_dir import func_a, func_b

# work_dir/__init__.py
from .foo import func_a
from .bar import func_b

#tests.py
from prj_dir.main import main_func

В таком случае скрипт python3 main.py работает корректно, но при попытке запуска тестов:
ModuleNotFoundError: No module named 'work_dir'

Если в main.py добавить точку в импорте:
from .work_dir import func_a, func_b то тесты проходят но скрипт не работает.