[gentoo-commits] proj/pms-test-suite:master commit in: pmstestsuite/library/

[Michał Górny]

commit:     e1e9de99067427288827b52c53a8de3871e930ef
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Tue Aug  2 18:12:15 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Tue Aug  2 18:12:15 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/pms-test-suite.git;a=commit;h=e1e9de99

Add .assertEqual()/.assertNotEqual() for the test cases.

---
 pmstestsuite/library/case.py |   28 ++++++++++++++++++++++++++++
 1 files changed, 28 insertions(+), 0 deletions(-)

diff --git a/pmstestsuite/library/case.py b/pmstestsuite/library/case.py
index a11bafe..6a37524 100644
--- a/pmstestsuite/library/case.py
+++ b/pmstestsuite/library/case.py
@@ -154,6 +154,34 @@ class TestCase(ABCObject):
 		if needle not in container:
 			raise AssertionError(msg)
 
+	def assertEqual(self, value, expect, msg):
+		"""
+		Assert that the value is equal to expected one.
+
+		@param value: the actual value
+		@type value: any
+		@param expect: the expected value
+		@type expect: any
+		@param msg: assertion description
+		@type msg: string
+		"""
+		if value != expect:
+			raise AssertionError(msg)
+
+	def assertNotEqual(self, value, unallowed, msg):
+		"""
+		Assert that the value is other than an unallowed one.
+
+		@param value: the actual value
+		@type value: any
+		@param expect: the unallowed value
+		@type expect: any
+		@param msg: assertion description
+		@type msg: string
+		"""
+		if value == unallowed:
+			raise AssertionError(msg)
+
 	@abstractmethod
 	def check_result(self, pm):
 		"""