PyContract检查导入模块中的函数

时间:2012-10-22 02:11:40

标签: python design-by-contract

我正在使用PyContract(而不是PyContracts)来实现我的遗传算法框架的合同设计。

主要模块是GA.py,它从crossover.py调用一个函数(我们称之为foocross)。我为foocross编写了前置条件和后置条件表达式。

现在,我希望在GA.py中运行main时检查这些条件。我在foocross中引入了一个错误的前置条件,只是为了测试合同是否被检查,但似乎它们不是。我如何确保正在检查合同?

以下是我的代码的最小部分:

# GA.py
def main():
    # some code
    crossover.foocross(params)
    # some more code

if __name__ == "__main__":
    main()

#crossover.py
def injectionco(p1, p2, chrom):
    """
        pre:
            isinstance(p1, int) # this should raise an error. p1 is never an int
            isinstance(p2, list)
            isinstance(chrom, int)
            forall(p1, lambda elem: any(isinstance(elem, i.__class__) for i in p2))
            forall(p2, lambda elem: any(isinstance(elem, i.__class__) for i in p1))
            len(p1) == len(p2)

        post:
            p1 is __old__.p1
            p2 is __old__.p2
            p1 == __old__.p1
            p2 == __old__.p2
            isinstance(__return__, p1.__class__)    # returns an individual
            len(__return__) == len(p1)
            id(__return__) not in [id(p1), id(p2)]
            forall(__return__, lambda elem: __return__.count(elem) == 1)
        """

    #code
    return #something
PS:这对我来说是一个非常困难的问题,因为我必须删除大量的代码(因为框架非常复杂),而且我是按合同设计的新手。因此,如果我错过了一些有助于回答这个问题的信息,请询问,我很乐意提供更多信息:)

2 个答案:

答案 0 :(得分:0)

the example,您似乎需要做的就是:

# this needs to be there unconditionally.
import contract
contract.checkmod(__name__)

if __name__ == '__main__':
   import doctest, GA
   doctest.testmod(GA)

但是,下面的无效(因为在import GA内,contract.checkmod()不会发生):

if __name__ == '__main__':
   import contract
   contract.checkmod(__name__)
   import doctest, GA
   doctest.testmod(GA)

答案 1 :(得分:0)

还有另一种解决方案。对于大型代码库,编辑多个文件以启用/禁用每个模块的合同检查将非常繁琐。我能够从我的主模块调用所有合同检查,如下所示:

#GA.py

import crossover # and all other required modules
import contract

for mod in [crossover]: # put all other imported modules in that list
    contract.checkmod(mod)