Prettier output

Change-Id: Iab72262741fcbe85e19f25e321ee80d06d81f7c2
This commit is contained in:
Wes McKinney committed 2018-12-26 14:41:01 -06:00
1 parent a00b51c704
commit d9a88700dd
1 file changed
+7 -6
+7 -6
View File
@@ -54,20 +54,21 @@ class PandasConversionsFromArrow(PandasConversionsBase):
class ToPandasStrings(object):
params = ((100, 10000, 100000, 500000),)
param_names = ('uniqueness', 'total')
params = ((0.001, 0.01, 0.1, 0.5), (1000000,))
string_length = 25
num_elements = 1000 * 1000
def setup(self, nunique):
def setup(self, uniqueness, total):
nunique = int(total * uniqueness)
unique_values = [tm.rands(self.string_length) for i in range(nunique)]
values = unique_values * (self.num_elements // nunique)
values = unique_values * (total // nunique)
self.arr = pa.array(values, type=pa.string())
self.table = pa.Table.from_arrays([self.arr], ['f0'])
def time_to_pandas_dedup(self, nunique):
def time_to_pandas_dedup(self, *args):
self.arr.to_pandas()
def time_to_pandas_no_dedup(self, nunique):
def time_to_pandas_no_dedup(self, *args):
self.arr.to_pandas(deduplicate_objects=False)