From 8f481c478b9aac37a907eeb58b5e0804a7f4d1a0 Mon Sep 17 00:00:00 2001 From: Tim Graham Date: Tue, 25 Nov 2014 18:22:19 -0500 Subject: Added unittest aliases. --- six.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'six.py') diff --git a/six.py b/six.py index a8cdb22..303fd2d 100644 --- a/six.py +++ b/six.py @@ -610,6 +610,9 @@ if PY3: import io StringIO = io.StringIO BytesIO = io.BytesIO + _assertCountEqual = "assertCountEqual" + _assertRaisesRegex = "assertRaisesRegex" + _assertRegex = "assertRegex" else: def b(s): return s @@ -625,10 +628,25 @@ else: iterbytes = functools.partial(itertools.imap, ord) import StringIO StringIO = BytesIO = StringIO.StringIO + _assertCountEqual = "assertItemsEqual" + _assertRaisesRegex = "assertRaisesRegexp" + _assertRegex = "assertRegexpMatches" _add_doc(b, """Byte literal""") _add_doc(u, """Text literal""") +def assertCountEqual(self, *args, **kwargs): + return getattr(self, _assertCountEqual)(*args, **kwargs) + + +def assertRaisesRegex(self, *args, **kwargs): + return getattr(self, _assertRaisesRegex)(*args, **kwargs) + + +def assertRegex(self, *args, **kwargs): + return getattr(self, _assertRegex)(*args, **kwargs) + + if PY3: exec_ = getattr(moves.builtins, "exec") -- cgit v1.2.1