class TestCase(unittest.TestCase):
"""Base class for test cases for kgb."""
ws_re = re.compile(r'\s+')
def setUp(self):
self.agency = SpyAgency()
self.orig_class_do_math = MathClass.class_do_math
def tearDown(self):
MathClass.class_do_math = self.orig_class_do_math
self.agency.unspy_all()
def shortDescription(self):
"""Return the description of the current test.
This changes the default behavior to replace all newlines with spaces,
allowing a test description to span lines. It should still be kept
short, though.
Returns:
bytes:
The description of the test.
"""
doc = self._testMethodDoc
if doc is not None:
doc = doc.split('\n\n', 1)[0]
doc = self.ws_re.sub(' ', doc).strip()
return doc
class TestCase(unittest.TestCase):
"""Base class for test cases for kgb."""
ws_re = re.compile(r'\s+')
def setUp(self):
self.agency = SpyAgency()
self.orig_class_do_math = MathClass.class_do_math
def tearDown(self):
MathClass.class_do_math = self.orig_class_do_math
self.agency.unspy_all()
def shortDescription(self):
"""Return the description of the current test.
This changes the default behavior to replace all newlines with spaces,
allowing a test description to span lines. It should still be kept
short, though.
Returns:
bytes:
The description of the test.
"""
doc = self._testMethodDoc
if doc is not None:
doc = doc.split('\n\n', 1)[0]
doc = self.ws_re.sub(' ', doc).strip()
return doc
def make_func(self, code_str, func_name='func'):
"""Return a new function, created by the supplied Python code.
This is used to create functions with signatures that depend on a
specific version of Python, and would generate syntax errors on
earlier versions.
Args:
code_str (unicode):
The Python code used to create the function.
func_name (unicode, optional):
The expected name of the function.
Returns:
callable:
The resulting function.
Raises:
Exception:
There was an error with the supplied code.
"""
scope = {}
exec(textwrap.dedent(code_str), scope)
return scope[func_name]
class BaseTestCase(unittest.TestCase):
"""Base class for test cases for kgb."""
def setUp(self):
self.agency = SpyAgency()
self.orig_class_do_math = MathClass.class_do_math
def tearDown(self):
MathClass.class_do_math = self.orig_class_do_math
self.agency.unspy_all()
class BaseTestCase(unittest.TestCase):
"""Base class for test cases for kgb."""
def setUp(self):
self.agency = SpyAgency()
self.orig_class_do_math = MathClass.class_do_math
def tearDown(self):
MathClass.class_do_math = self.orig_class_do_math
self.agency.unspy_all()