python - 独立运行python包子模块进行自测试；导入路径困惑

Question

我经常在模块的底部编写自测试代码，即

if __name__ == '__main__':

.
.
.

我想将其保留在模块中，这样如果我修改它，我仍然可以对其进行自检。该模块是包的一部分。所以存在包间引用需要解决；但如果我导入包与独立运行模块，这些问题的解决方式会有所不同。

我最终在我的模块顶部遇到了这样的困惑，这肯定很丑陋，而且可能不是“Pythonic”:

if __name__ == '__main__':

    from CovSample import CovSample
    from ArrayByRow import ArrayByRow    
else:

    from CEOpt import CovSample
    from CEOpt import ArrayByRow

这有效 - 如果我导入包 CEOpt - else-branch 引用有效，如果我独立运行 - 直接模块名称导入有效。但这并不漂亮，我想要一个用于包间调用的导入语句，该语句仍然可以在独立模块测试中工作。这可能吗？

Answer 1

It seems没有适用于这两种情况的通用方法，__name__=='__main__' 和 __name__!='__main__'。

我所做的如下:在每个子模块内定义一个 Test() 函数(因此 CEOpt.CovSample.Test() 和 CEOpt.ArrayByRow。测试())。然后创建一个 CEOpt/__main__.py 文件，这是 CEOpt 包中您将直接“运行”的唯一文件(该特定文件名确保它是当您在 shell 中输入 python -m CEOpt 时就会运行的那个)。此文件必须按名称显式导入 CEOpt，但至少现在只需在一个位置发生(其他文件可以使用带前缀点语法的相对导入，例如 from .CovArray import CovArray )。现在对 __main__.py 的逻辑进行编程，使其响应通过 sys.argv 传入的子命令，选择要执行的子模块的 Test() 函数运行。

例如，语法 python -m CEOpt test CovArray (或等效于 IPython 提示符下的 %run CEOpt/__main__ test CovArray)将触发 __main__ .py 调用 CEOpt.CovArray.Test()