[gentoo-commits] proj/gentoopm:master commit in: gentoopm/tests/

["Michał Górny" <mgorny@gentoo.org>]

commit:     ff335f908bfe971ee8961580e6933950c2ac3ef0
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Tue Jul 19 09:30:37 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Tue Jul 19 09:30:37 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoopm.git;a=commit;h=ff335f90

Tests: prepend all exceptions with PM name.

---
 gentoopm/tests/__init__.py |    8 ++++++--
 1 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/gentoopm/tests/__init__.py b/gentoopm/tests/__init__.py
index fbb773f..807af17 100644
--- a/gentoopm/tests/__init__.py
+++ b/gentoopm/tests/__init__.py
@@ -21,8 +21,12 @@ class PMTestCase(unittest.TestCase):
 	def _try(self, f, args):
 		try:
 			f(self, *args)
-		except AssertionError as e:
-			raise AssertionError('[%s] %s' % (self._pm.name, e.args[0]))
+		except Exception as e:
+			if isinstance(e.args[0], str):
+				raise e.__class__('[%s] %s' % (self._pm.name, e.args[0]),
+						*e.args[1:])
+			else:
+				raise
 
 	def assertEqual(self, *args):
 		self._try(unittest.TestCase.assertEqual, args)