From e6918187568dbd01842d8d1d2c808ce16a894239 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 21 Apr 2024 13:54:28 +0200 Subject: Adding upstream version 18.2.2. Signed-off-by: Daniel Baumann --- src/arrow/python/benchmarks/microbenchmarks.py | 45 ++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 src/arrow/python/benchmarks/microbenchmarks.py (limited to 'src/arrow/python/benchmarks/microbenchmarks.py') diff --git a/src/arrow/python/benchmarks/microbenchmarks.py b/src/arrow/python/benchmarks/microbenchmarks.py new file mode 100644 index 000000000..f8ba383c7 --- /dev/null +++ b/src/arrow/python/benchmarks/microbenchmarks.py @@ -0,0 +1,45 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License. + +import pyarrow.benchmark as pb + +from . import common + + +class PandasObjectIsNull(object): + size = 10 ** 5 + types = ('int', 'float', 'object', 'decimal') + + param_names = ['type'] + params = [types] + + def setup(self, type_name): + gen = common.BuiltinsGenerator() + if type_name == 'int': + lst = gen.generate_int_list(self.size) + elif type_name == 'float': + lst = gen.generate_float_list(self.size, use_nan=True) + elif type_name == 'object': + lst = gen.generate_object_list(self.size) + elif type_name == 'decimal': + lst = gen.generate_decimal_list(self.size) + else: + assert 0 + self.lst = lst + + def time_PandasObjectIsNull(self, *args): + pb.benchmark_PandasObjectIsNull(self.lst) -- cgit v1.2.3