python——将函数参数注释为特定模块

我有一个pytest夹具，可以导入特定的模块。这是必需的，因为导入模块非常昂贵，所以我们不希望在导入时（即在pytest收集期间）进行导入。这会产生如下代码：

@pytest.fixture
def my_module_fix():
    import my_module
    yield my_module

def test_something(my_module_fix):
    assert my_module_fix.my_func() = 5

我正在使用PyCharm，希望在测试中进行类型检查和自动完成。为了实现这一点，我必须以某种方式对my_module_fix参数的类型my_module单元

我不知道如何实现这一点。我只发现我可以注释my_module_fix作为一种类型types.ModuleType，但这还不够：它不是任何模块，总是my_module.