Skip to content

Commit 82788c4

Browse files
committed
[Benchmark] Adapt data generation of integer columns.
1 parent f36f51a commit 82788c4

File tree

1 file changed

+1
-15
lines changed

1 file changed

+1
-15
lines changed

benchmark/operators/gen.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -142,21 +142,7 @@ def gen_random_int_values(smallest :int, largest :int, num :int):
142142
if largest - smallest == num:
143143
return list(range(smallest, largest))
144144

145-
taken = set()
146-
counter = largest - num
147-
values = list()
148-
149-
for i in range(0, num):
150-
val = random.randrange(smallest, largest - num)
151-
if val in taken:
152-
values.append(counter)
153-
counter += 1
154-
else:
155-
taken.add(val)
156-
values.append(val)
157-
158-
assert len(values) == len(set(values))
159-
return values
145+
return random.sample(range(smallest, largest), k=num)
160146

161147

162148
#=======================================================================================================================

0 commit comments

Comments
 (0)