增加在Python中进行单元测试的类的日志记录详细程度

Question

I have a Python class that uses the logging module to provide some debug output:

File someclass.py:

import logging

class SomeClass:

    def do_stuff(self):
        # do some things
        logging.debug("I just did some stuff")
        # do some more stuff
        return True

I do unit testing on this class with the unittest module

File test_someclass.py

import unittest
from someclass import SomeClass

class SomeClassTests(unittest.TestCase):

    def test_do_stuff(self):
        obj = SomeClass()
        self.assertFalse(obj.do_stuff())

def main():
    unittest.main()

if __name__ == '__main__':
    main()

What I want to do is show the debug messages while I am running the unit tests. I tried to set the verbosity to debug from the unit test module:

import logging

# ....

def main():
    unittest.main()
    logging.basicConfig(level=logging.DEBUG)

This didn't work. What would be the way to achieve this? Even better would be enabling DEBUG verbosity for only one test.

UPDATE:

Apparently it works when running it from the Python shell, but not in PyDev (it probably uses a different test runner).

Answer 1

call unittest.main() from your main() .

def main():
    logging.basicConfig(level=logging.DEBUG)
    unittest.main()

My output shows:

DEBUG:root:I just did some stuff
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

The basic example on unittests docs at: https://docs.python.org/2/library/unittest.html shows the simple way of calling and running a unit test from main.

Answer 2

If you want to output debug messages on failures only, using nose test runner would be the easiest way to go since nose captures stdout and print it out on failures. It works out of the box:

$ nosetests test.py
F
======================================================================
FAIL: test_stuff (test.SomeClass)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/../../test.py", line 7, in test_stuff
    self.assertFalse(True)
AssertionError: True is not false
-------------------- >> begin captured stdout << ---------------------
I just did some stuff

--------------------- >> end captured stdout << ----------------------

----------------------------------------------------------------------
Ran 1 test in 0.001s

FAILED (failures=1)

where test.py contains:

from unittest import TestCase

class SomeClass(TestCase):
    def test_stuff(self):
        print "I just did some stuff"
        self.assertFalse(True)