def test_sorted_set(self):
writer = SortedSetWriter(str)
hzset = SortedSetHandle('myzset', str)
writer.append(hzset, 'spam', 3.25)
writer.append(hzset, 'eggs', 3.24)
self.assertEqual(SortedSet(hzset), ('eggs', 'spam'))
self.assertEqual(SortedSet(hzset, lte = 3.24), ('eggs',))
self.assertEqual(SortedSet.zrange(hzset, 0, 0), ('eggs',))
self.assertEqual(SortedSet.zfind(hzset, gt = 3.24), ('spam',))
def test_model_sorted_set(self):
writer = SortedSetWriter(Fighter)
hzset = SortedSetHandle('myzset', Fighter)
f1, f2, f3 = map(Fighter.by_id, [21, 33, 47])
writer.append(hzset, f1, 3.25)
writer.append(hzset, f2, 3.24)
writer.append(hzset, f3, 4)
self.assertEqual(SortedSet.zrange(hzset), (f2, f1, f3))
self.assertEqual(SortedSet.zrevrange(hzset, 0, 0), (f3,))
self.assertEqual(SortedSet(hzset, gt = 3.24), (f1, f3))
