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)