def test_keys_without_prefix(self):
ws = WorkloadSettings(items=10 ** 3, workers=20, working_set=100,
working_set_access=100, working_set_moving_docs=0,
key_fmtr='decimal')
keys = set()
for worker in range(ws.workers):
for key in docgen.SequentialKey(sid=worker, ws=ws, prefix=''):
keys.add(key.string)
expected = [docgen.Key(number=i, prefix='', fmtr='decimal').string
for i in range(ws.items)]
self.assertEqual(sorted(keys), expected)
def test_seq_key_generator(self):
ws = WorkloadSettings(items=10 ** 5, workers=25, working_set=100,
working_set_access=100, working_set_moving_docs=0,
key_fmtr='decimal')
keys = []
for worker in range(ws.workers):
generator = docgen.SequentialKey(worker, ws, prefix='test')
keys += [key.string for key in generator]
expected = [docgen.Key(number=i, prefix='test', fmtr='decimal').string
for i in range(ws.items)]
self.assertEqual(sorted(keys), expected)
def test_incompressible_docs(self):
size = 15 * 1024
generator = docgen.IncompressibleString(avg_size=size)
doc = generator.next(key=docgen.Key(number=0, prefix='', fmtr=''))
self.assertEqual(len(doc), size)
