Rich comparison tests

1 files changed, 177 insertions, 0 deletions

diff --git a/Lib/test/test_richcmp.py b/Lib/test/test_richcmp.py
new file mode 100644
index 0000000..9c05e98
--- /dev/null
+++ b/Lib/test/test_richcmp.py
@@ -0,0 +1,177 @@
+# Tests for rich comparisons
+
+from test_support import TestFailed
+
+class Number:
+
+    def __init__(self, x):
+        self.x = x
+
+    def __lt__(self, other):
+        return self.x < other
+
+    def __le__(self, other):
+        return self.x <= other
+
+    def __eq__(self, other):
+        return self.x == other
+
+    def __ne__(self, other):
+        return self.x != other
+
+    def __gt__(self, other):
+        return self.x > other
+
+    def __ge__(self, other):
+        return self.x >= other
+
+    def __cmp__(self, other):
+        raise TestFailed, "Number.__cmp__() should not be called"
+
+    def __repr__(self):
+        return "Number(%s)" % repr(self.x)
+
+class Vector:
+
+    def __init__(self, data):
+        self.data = data
+
+    def __len__(self):
+        return len(self.data)
+
+    def __getitem__(self, i):
+        return self.data[i]
+
+    def __setitem__(self, i, v):
+        self.data[i] = v
+
+    def __hash__(self):
+        raise TypeError, "Vectors cannot be hashed"
+
+    def __nonzero__(self):
+        raise TypeError, "Vectors cannot be used in Boolean contexts"
+
+    def __cmp__(self, other):
+        raise TestFailed, "Vector.__cmp__() should not be called"
+
+    def __repr__(self):
+        return "Vector(%s)" % repr(self.data)
+
+    def __lt__(self, other):
+        return Vector([a < b for a, b in zip(self.data, self.__cast(other))])
+
+    def __le__(self, other):
+        return Vector([a <= b for a, b in zip(self.data, self.__cast(other))])
+
+    def __eq__(self, other):
+        return Vector([a == b for a, b in zip(self.data, self.__cast(other))])
+
+    def __ne__(self, other):
+        return Vector([a != b for a, b in zip(self.data, self.__cast(other))])
+
+    def __gt__(self, other):
+        return Vector([a > b for a, b in zip(self.data, self.__cast(other))])
+
+    def __ge__(self, other):
+        return Vector([a >= b for a, b in zip(self.data, self.__cast(other))])
+
+    def __cast(self, other):
+        if isinstance(other, Vector):
+            other = other.data
+        if len(self.data) != len(other):
+            raise ValueError, "Cannot compare vectors of different length"
+        return other
+
+operators = "<", "<=", "==", "!=", ">", ">="
+opmap = {}
+for op in operators:
+    opmap[op] = eval("lambda a, b: a %s b" % op)
+
+def testvector():
+    a = Vector(range(2))
+    b = Vector(range(3))
+    for op in operators:
+        try:
+            opmap[op](a, b)
+        except ValueError:
+            pass
+        else:
+            raise TestFailed, "a %s b for different length should fail" % op
+    a = Vector(range(5))
+    b = Vector(5 * [2])
+    for op in operators:
+        print "%23s %-2s %-23s -> %s" % (a, op, b, opmap[op](a, b))
+        print "%23s %-2s %-23s -> %s" % (a, op, b.data, opmap[op](a, b.data))
+        print "%23s %-2s %-23s -> %s" % (a.data, op, b, opmap[op](a.data, b))
+        try:
+            if opmap[op](a, b):
+                raise TestFailed, "a %s b shouldn't be true" % op
+            else:
+                raise TestFailed, "a %s b shouldn't be false" % op
+        except TypeError:
+            pass
+
+def testop(a, b, op):
+    try:
+        ax = a.x
+    except AttributeError:
+        ax = a
+    try:
+        bx = b.x
+    except AttributeError:
+        bx = b
+    opfunc = opmap[op]
+    realoutcome = opfunc(ax, bx)
+    testoutcome = opfunc(a, b)
+    if realoutcome != testoutcome:
+        print "Error for", a, op, b, ": expected", realoutcome,
+        print "but got", testoutcome
+##    else:
+##        print a, op, b, "-->", testoutcome # and "true" or "false"
+
+def testit(a, b):
+    testop(a, b, "<")
+    testop(a, b, "<=")
+    testop(a, b, "==")
+    testop(a, b, "!=")
+    testop(a, b, ">")
+    testop(a, b, ">=")
+
+def basic():
+    for a in range(3):
+        for b in range(3):
+            testit(Number(a), Number(b))
+            testit(a, Number(b))
+            testit(Number(a), b)
+
+def tabulate(c1=Number, c2=Number):
+    for op in operators:
+        opfunc = opmap[op]
+        print
+        print "operator:", op
+        print
+        print "%9s" % "",
+        for b in range(3):
+            b = c2(b)
+            print "| %9s" % b,
+        print "|"
+        print '----------+-' * 4
+        for a in range(3):
+            a = c1(a)
+            print "%9s" % a,
+            for b in range(3):
+                b = c2(b)
+                print "| %9s" % opfunc(a, b),
+            print "|"
+        print '----------+-' * 4
+    print
+    print '*' * 50
+                
+def main():
+    basic()
+    tabulate()
+    tabulate(c1=int)
+    tabulate(c2=int)
+    testvector()
+
+main()