def test_get_key(self): h = db.Hash('h1') h['hk1'] = 'v1' l = db.List('l1') l.append('i1') s = db.Set('s1') s.add('k1') zs = db.ZSet('z1') zs.add({'i1': 1., 'i2': 2.}) h_db = db.get_key('h1') self.assertTrue(isinstance(h_db, Hash)) self.assertEqual(h_db['hk1'], b'v1') l_db = db.get_key('l1') self.assertTrue(isinstance(l_db, List)) self.assertEqual(l_db[0], b'i1') s_db = db.get_key('s1') self.assertTrue(isinstance(s_db, Set)) self.assertEqual(s_db.members(), set((b'k1', ))) z_db = db.get_key('z1') self.assertTrue(isinstance(z_db, ZSet)) self.assertEqual(z_db.score('i1'), 1.)
def test_combining(self): self.set2 = db.Set('my-set2') self.set.add(1, 2, 3, 4) self.set2.add(3, 4, 5, 6) self.assertEqual(self.set | self.set2, set([b'1', b'2', b'3', b'4', b'5', b'6'])) self.assertEqual(self.set & self.set2, set([b'3', b'4'])) self.assertEqual(self.set - self.set2, set([b'1', b'2'])) self.assertEqual(self.set2 - self.set, set([b'5', b'6']))
def test_combine_store(self): self.set2 = db.Set('my-set2') self.set.add(1, 2, 3, 4) self.set2.add(3, 4, 5, 6) s3 = self.set.unionstore('my-set3', self.set2) self.assertEqual(s3.members(), set([b'1', b'2', b'3', b'4', b'5', b'6'])) s3 = self.set.interstore('my-set3', self.set2) self.assertEqual(s3.members(), set([b'3', b'4'])) s3 = self.set.diffstore('my-set3', self.set2) self.assertEqual(s3.members(), set([b'1', b'2'])) self.set |= self.set2 self.assertEqual(sorted(self.set), [b'1', b'2', b'3', b'4', b'5', b'6']) s4 = db.Set('my-set4') s4.add('1', '3') s3 &= s4 self.assertEqual(s3.members(), set([b'1']))
def setUp(self): super(TestSet, self).setUp() self.set = db.Set('my-set')
def test_as_set(self): self.set.add('foo', 'bar', 'baz') self.assertEqual(self.set.as_set(True), set(('foo', 'bar', 'baz'))) self.assertEqual(db.Set('test').as_set(), set())