Merge pull request #478 from opensourcerouting/test-extension

tests: allow to use exit_cleanly as test method
This commit is contained in:
Donald Sharp 2017-05-09 11:55:54 -04:00 committed by GitHub
commit 7a78dea34d

View File

@ -87,7 +87,8 @@ class _TestMultiOut(object):
def _add_test(cls, method, *args, **kwargs):
if 'tests' not in dir(cls):
setattr(cls,'tests',[])
cls._add_test(cls._exit_cleanly)
if method is not cls._exit_cleanly:
cls._add_test(cls._exit_cleanly)
def matchfunction(self):
method(self, *args, **kwargs)