import dependency_injector.containers as dic
import dependency_injector.providers as dip


class UnitOfWork:
    ...


class TestUnitOfWork(UnitOfWork):
    ...


class ApplicationContainer(dic.DeclarativeContainer):
    unit_of_work_class = dip.Callable(UnitOfWork)
    uow = dip.Factory(unit_of_work_class)


class TestApplicationContainer(ApplicationContainer):
    unit_of_work_class = dip.Callable(TestUnitOfWork)


container = TestApplicationContainer()
print(container.uow())
with container.unit_of_work_class.override(TestUnitOfWork):
    print(container.uow())