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())