def test_reducer(self):
red = TokyoCabinetReducer()
output = red(zip('abcde', '12345'))
fn = mkstemp()[1]
fo = open(fn, 'wb')
fo.writelines(v for k, v in output)
fo.close()
db = HDB()
db.open(fn, HDBOREADER)
self.assertEqual(list(db.iteritems()),
[('a', '1'), ('b', '2'), ('c', '3'), ('d', '4'), ('e', '5')])
db.close()
os.remove(fn)
def test_reducer(self):
red = TokyoCabinetReducer()
output = red(zip('abcde', '12345'))
fn = mkstemp()[1]
fo = open(fn, 'wb')
fo.writelines(v for k, v in output)
fo.close()
db = HDB()
db.open(fn, HDBOREADER)
self.assertEqual(list(db.iteritems()), [('a', '1'), ('b', '2'),
('c', '3'), ('d', '4'),
('e', '5')])
db.close()
os.remove(fn)
def test_default(self):
proc = TokyoCabinetFactory()
self.assertEqual(proc('k1', ['v1']), None)
self.assertEqual(proc('k2', ['v2', 'v3']), None)
chunks = proc.close()
fn = mkstemp()[1]
fo = open(fn, 'wb')
for chk in chunks:
self.assertTrue(len(chk) <= proc.chunksize)
fo.write(chk)
fo.close()
db = HDB()
db.open(fn, HDBOREADER)
self.assertEqual(list(db.iteritems()), [('k1', 'v1'), ('k2', 'v3')])
db.close()
os.remove(fn)
def test_default(self):
proc = TokyoCabinetFactory()
self.assertEqual(proc('k1', ['v1']), None)
self.assertEqual(proc('k2', ['v2', 'v3']), None)
chunks = proc.close()
fn = mkstemp()[1]
fo = open(fn, 'wb')
for chk in chunks:
self.assertTrue(len(chk) <= proc.chunksize)
fo.write(chk)
fo.close()
db = HDB()
db.open(fn, HDBOREADER)
self.assertEqual(list(db.iteritems()), [('k1', 'v1'), ('k2', 'v3')])
db.close()
os.remove(fn)
