Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Commit 5b79b1d

Browse files
committed
Minor fix in generating data for some perf.tests
1 parent 9bae742 commit 5b79b1d

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

sdc/tests/tests_perf/test_perf_series.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,9 +56,10 @@ def setUpClass(cls):
5656

5757
def _test_series(self, pyfunc, name):
5858
input_data = test_global_input_data_float64
59+
full_input_data_length = sum(len(i) for i in input_data)
5960
hpat_func = sdc.jit(pyfunc)
6061
for data_length in self.total_data_length:
61-
data, = perf_data_gen_fixed_len(input_data, data_length, 1)
62+
data = perf_data_gen_fixed_len(input_data, full_input_data_length, data_length)
6263
test_data = pd.Series(data)
6364

6465
compile_results = calc_compilation(pyfunc, test_data, iter_number=self.iter_number)

sdc/tests/tests_perf/test_perf_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def perf_data_gen(tmpl, max_item_len, max_bytes_size):
115115
max_item_len: length (in elements) of resulted string in an element of the result array
116116
max_bytes_size: maximum size in bytes of the return data
117117
118-
return: list of iterable data
118+
return: list of data
119119
"""
120120
result = []
121121
while get_size(result) < max_bytes_size:
@@ -135,7 +135,7 @@ def perf_data_gen_fixed_len(tmpl, max_item_len, max_obj_len):
135135
max_item_len: length (in elements) of resulted string in an element of the result array
136136
max_obj_len: maximum length of the return data
137137
138-
return: list of iterable data
138+
return: list of data
139139
"""
140140
result = []
141141
while len(result) < max_obj_len:

0 commit comments

Comments
 (0)