[Zope-CVS] CVS: Products/ZCTextIndex/tests - testSetOps.py:1.3

[Tim Peters]

Update of /cvs-repository/Products/ZCTextIndex/tests
In directory cvs.zope.org:/tmp/cvs-serv16580/tests

Modified Files:
	testSetOps.py 
Log Message:
New testMany() -- a meatier test of the mass set ops.


=== Products/ZCTextIndex/tests/testSetOps.py 1.2 => 1.3 ===
                     self.assertEqual(expected, list(got.items()))
 
+    def testMany(self):
+        import random
+        N = 15  # number of IIBTrees to feed in
+        L = []
+        commonkey = N * 1000
+        allkeys = {commonkey: 1}
+        for i in range(N):
+            t = IIBTree()
+            t[commonkey] = i
+            for j in range(N-i):
+                key = i + j
+                allkeys[key] = 1
+                t[key] = N*i + j
+            L.append((t, i+1))
+        random.shuffle(L)
+        allkeys = allkeys.keys()
+        allkeys.sort()
+
+        # Test the union.
+        expected = []
+        for key in allkeys:
+            sum = 0
+            for t, w in L:
+                if t.has_key(key):
+                    sum += t[key] * w
+            expected.append((key, sum))
+        # print 'union', expected
+        got = mass_weightedUnion(L)
+        self.assertEqual(expected, list(got.items()))
+
+        # Test the intersection.
+        expected = []
+        for key in allkeys:
+            sum = 0
+            for t, w in L:
+                if t.has_key(key):
+                    sum += t[key] * w
+                else:
+                    break
+            else:
+                # We didn't break out of the loop so it's in the intersection.
+                expected.append((key, sum))
+        # print 'intersection', expected
+        got = mass_weightedIntersection(L)
+        self.assertEqual(expected, list(got.items()))
+
 def test_suite():
     return makeSuite(TestSetOps)