如何在python中撤消“assert”?

时间:2014-02-07 04:04:46

标签: python assert

在早期的功能中,我们称之为a,我必须确保a函数不会产生太小的答案(让我们说,答案必须大于10) )。 但是稍后,在第二个函数b(使用函数a作为其输入之一)中,如果被修改的相同答案小于之前的assert,则可以。声明。

有办法做到这一点吗?

当函数b产生的答案太小时,我试图在变量中保存答案的小小,但是当我运行doctest时我仍然遇到断言错误。

def grade_on_project(student_project_score, max_score, overall_score):
    project_grade = student_project_score / max_project_score
    assert project_grade > 0.6 # (student fails the class if any of their scores on a project are too low)
    overall_score +=student_project_score
    return overall_score

def who_fails_first(operation, person1, person2)
    if (operation(person1, max_score, person1) <= 150 and (operation(person2, max_score, overall_score) > 150:
        print(student 1 failed)
    if (operation(person2, max_score, overall_score) <= 150 and (operation(person1, max_score, overall_score) > 150:
        print(student 2 failed)

 who_fails_first(grade_on_project, 5, 9)

1 个答案:

答案 0 :(得分:2)

断言应用于实现某段代码的逻辑。为什么?因为你不能保证它们真的被提升了!

$python -c 'assert False'
Traceback (most recent call last):
  File "<string>", line 1, in <module>
AssertionError
$python -O -c 'assert False'   # doesn't raise an error!
$

-O选项用于“优化”。但它也删除了所有断言

assert应该用于 debug ,检查在编写良好的程序中永远不会发生的条件。

如果您想以某种方式处理特殊输入,则应使用例外

def grade_on_project(student_project_score, max_score, overall_score):
    project_grade = student_project_score / max_project_score

    if project_grade > 0.6:
        raise ValueError('Student fails if at least one project grade is too low.')

    overall_score +=student_project_score
    return overall_score

或者您可以编写自定义异常,例如:

class StudentFailed(ValueError):
    def __init__(self, message, grade):
        super(StudentFailed, self).__init__(message)
        self.grade = grade

并将if更改为:

if project_grade > 0.6:
    raise StudentFailed('One project has grade too low.', project_grade)

然后,当您捕获异常时,您仍然可以通过异常的grade属性访问使其失败的等级。

如果您不想在特定情况下引发异常,可以添加一个参数,例如low_grade_is_ok,并将该函数修改为:

def grade_on_project(student_project_score, max_score,
                     overall_score, low_grade_is_ok=False):
    project_grade = student_project_score / max_project_score

    if not low_grade_is_ok and project_grade > 0.6:
        raise ValueError('Student fails if at least one project grade is too low.')

    overall_score +=student_project_score
    return overall_score

然后在你不想引发异常的函数中,可以用low_grade_is_ok=True调用它,在这种情况下不会引发任何异常。