New MATCH() argument matcher, which allows using a function

to match an argument generically.  E.g. MATCH(lambda x: x > 10)

1 files changed, 11 insertions, 1 deletions

diff --git a/mocker.py b/mocker.py
index e546266..ee19523 100644
--- a/mocker.py
+++ b/mocker.py
@@ -18,7 +18,8 @@ if sys.version_info < (2, 4):
     from sets import Set as set # pragma: nocover
 
 
-__all__ = ["Mocker", "expect", "IS", "CONTAINS", "IN", "ANY", "ARGS", "KWARGS"]
+__all__ = ["Mocker", "expect", "IS", "CONTAINS", "IN", "MATCH",
+           "ANY", "ARGS", "KWARGS"]
 
 
 __author__ = "Gustavo Niemeyer <gustavo@niemeyer.net>"
@@ -1350,6 +1351,15 @@ class IN(SpecialArgument):
         return other in self.object
 
 
+class MATCH(SpecialArgument):
+
+    def matches(self, other):
+        return bool(self.object(other))
+
+    def __eq__(self, other):
+        return type(other) == type(self) and self.object is other.object
+
+
 def match_params(args1, kwargs1, args2, kwargs2):
     """Match the two sets of parameters, considering special parameters."""