CODEWITHSUNDEEP
pythonunit-testingpytestassertion
cool77
cool77

Reputation: 1136

print success messages for asserts in python

I am using assert in python. Every time an assert fails I get the failure message which I would have put there to be printed. I was wondering if there is a way to print a custom success message when the assert condition passes?

I am using py.test framework.

for instance: 

assert self.clnt.stop_io()==1, "IO stop failed"

for the above assert I get message "IO stop failed" if assert fails but I am looking to have "IO stop succeeded" if assert passes. something like this:

 assert self.clnt.stop_io()==1, "IO stop failed", "IO stop succeeded"

Upvotes: 8

Views: 23668

Answers (5)

JeffCharter
JeffCharter

Reputation: 1549

This has been added to the pytest module v5 as the hook pytest_assertion_pass

in your conftest.py

def pytest_assertion_pass(item, lineno, orig, expl):
    log.info("asserting that {}, {}, {}, {}".format(item, lineno, orig, expl))

and in pytest.ini

enable_assertion_pass_hook = true

Upvotes: 7

Eugene
Eugene

Reputation: 1895

I would combine @dashingdw and @Reblochon Masque solution

def my_assert(condition, fail_str, suc_str):
    assert condition, fail_str
    print suc_str

From my point of view this is less "dangerous" and you don't need to insert an additional print line every time.

Upvotes: 3

cdarke
cdarke

Reputation: 44434

Write a simple helper function:

def myfunc(msg='assert OK'):
    print msg
    return True

Include that in the condition on the right-side, after an and:

assert self.clnt.stop_io()==1 and myfunc("IO stop succeeded"), "IO stop failed"

Upvotes: 8

WilliamComputerScience
WilliamComputerScience

Reputation: 87

If you are really feeling adventurous, you can just write your own extensible assert wrapper - with which you can count the number of asserts failed or add functionality like testing in quiet or verbose modes eg:

def assert(condition, fail_str, suc_str):
     if condition:
          print fail_str
     else:
          num_tests_passed = num_tests_passed + 1
          if verbose_enabled:
              print suc_str

Upvotes: 2

Reblochon Masque
Reblochon Masque

Reputation: 36732

Yes, the simplest is surely to place a print below the assert:

assert self.clnt.stop_io()==1, "IO stop failed"
print("IO stop passed at location ==1")

Upvotes: 14

Related Questions