fix itertools.production usage
diff --git a/data_generator.py b/data_generator.py
index 3692389..019b861 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -7,7 +7,7 @@
from storage_api import create_storage
-types = ["GA", "master"] + [pet_generate(2, '-') for _ in range(10)]
+types = ["GA", "master"] + [pet_generate(2, '-') for _ in range(2)]
random.shuffle(types)
tp = itertools.cycle(types)
@@ -16,14 +16,15 @@
is_sync = ["s", "a"]
storage = create_storage(sys.argv[1])
+combinations = list(itertools.product(op_type, is_sync, sz))
for i in range(30):
row = {"build_id": pet_generate(2, " "),
"type": next(tp),
"iso_md5": uuid.uuid4().get_hex()}
- for sz, op_type, is_sync in itertools.product(op_type, is_sync, sz):
- row[" ".join([sz, op_type, is_sync])] = (random.random(),
- random.random() / 5)
+ for sz, op_type, is_sync in combinations:
+ row[" ".join([sz, op_type, is_sync])] = (random.random() * 100,
+ random.random() * 5)
storage.store(row)