def test_dup_reader_and_writer(self):
def key_value_gen():
for i in range(3):
yield 789, "value%d" % (i * i)
writer = Writer('./test_rw_dup', int_key=True, dup=True)
writer.drop()
writer.put(123, 'bar')
writer.put(456, 'spam')
writer.mput({123: "bar1", 456: "spam1"})
writer.mput(key_value_gen())
reader = DupReader('./test_rw_dup', int_key=True)
values = reader.get(123)
self.assertEqual(list(values), ['bar', 'bar1'])
values = reader.get(456)
self.assertEqual(list(values), ['spam', 'spam1'])
values = reader.get(789)
self.assertEqual(list(values), ['value0', 'value1', 'value4'])
def test_dup_reader_and_writer(self):
def key_value_gen():
for i in range(3):
yield 'fixed', "value%d" % (i * i)
writer = Writer('./test_rw_dup', dup=True, encode_fn=dumps)
writer.drop()
writer.put('foo', 'bar')
writer.put('egg', 'spam')
writer.mput({"foo": "bar1", "egg": "spam1"})
writer.mput(key_value_gen())
reader = DupReader('./test_rw_dup', decode_fn=loads)
values = reader.get('foo')
self.assertEqual(list(values), ['bar', 'bar1'])
values = reader.get('egg')
self.assertEqual(list(values), ['spam', 'spam1'])
values = reader.get('fixed')
self.assertEqual(list(values), ['value0', 'value1', 'value4'])
def test_dup_reader_and_writer(self):
def key_value_gen():
for i in range(3):
yield 'fixed', "value%d" % (i * i)
writer = Writer('./test_rw_dup', dup=True,
encode_fn=dumps)
writer.drop()
writer.put('foo', 'bar')
writer.put('egg', 'spam')
writer.mput({"foo": "bar1", "egg": "spam1"})
writer.mput(key_value_gen())
reader = DupReader('./test_rw_dup',
decode_fn=loads)
values = reader.get('foo')
self.assertEqual(list(values), ['bar', 'bar1'])
values = reader.get('egg')
self.assertEqual(list(values), ['spam', 'spam1'])
values = reader.get('fixed')
self.assertEqual(list(values), ['value0', 'value1', 'value4'])
