FM / testing

/usr/local/lib64/python3.6/site-packages/pyarrow/include/arrow/testing/

/usr/local/lib64/python3.6/site-packages/pyarrow/include/arrow/testing

Name	Size	Mode	Actions
async_test_util.h	1539	0644	edit dl rm
executor_util.h	1885	0644	edit dl rm
extension_type.h	5007	0644	edit dl rm
future_util.h	6246	0644	edit dl rm
generator.h	8645	0644	edit dl rm
gtest_common.h	4207	0644	edit dl rm
gtest_compat.h	1299	0644	edit dl rm
gtest_util.h	28752	0644	edit dl rm
json_integration.h	4204	0644	edit dl rm
matchers.h	8781	0644	edit dl rm
pch.h	1204	0644	edit dl rm
random.h	21542	0644	edit dl rm
uniform_real.h	2969	0644	edit dl rm
util.h	6817	0644	edit dl rm
visibility.h	1548	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pyarrow/include/arrow/testing/async_test_util.h (1539B)

#pragma once

#include <atomic>
#include <memory>

#include "arrow/util/async_generator.h"
#include "arrow/util/future.h"

namespace arrow {
namespace util {

template <typename T>
class TrackingGenerator {
 public:
  explicit TrackingGenerator(AsyncGenerator<T> source)
      : state_(std::make_shared<State>(std::move(source))) {}

Future<T> operator()() {
    state_->num_read++;
    return state_->source();
  }

int num_read() { return state_->num_read.load(); }

private:
  struct State {
    explicit State(AsyncGenerator<T> source) : source(std::move(source)), num_read(0) {}

AsyncGenerator<T> source;
    std::atomic<int> num_read;
  };

std::shared_ptr<State> state_;
};

}  // namespace util
}  // namespace arrow

// 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. #pragma once #include #include #include "arrow/util/async_generator.h" #include "arrow/util/future.h" namespace arrow { namespace util { template class TrackingGenerator { public: explicit TrackingGenerator(AsyncGenerator source) : state_(std::make_shared(std::move(source))) {} Future operator()() { state_->num_read++; return state_->source(); } int num_read() { return state_->num_read.load(); } private: struct State { explicit State(AsyncGenerator source) : source(std::move(source)), num_read(0) {} AsyncGenerator source; std::atomic num_read; }; std::shared_ptr state_; }; } // namespace util } // namespace arrow