base_fixture.hpp

/*
 * Copyright (c) 2022-2023, NVIDIA CORPORATION.
 *
 * Licensed 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 <rmm/cuda_stream_view.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
 
#include <cudf_test/cudf_gtest.hpp>
 
namespace cuspatial {
namespace test {
 
class RMMResourceMixin {
  rmm::mr::device_memory_resource* _mr{rmm::mr::get_current_device_resource()};
  rmm::cuda_stream_view _stream{rmm::cuda_stream_default};
 
 public:
  rmm::mr::device_memory_resource* mr() { return _mr; }
 
  rmm::cuda_stream_view stream() { return _stream; }
};
 
class BaseFixture : public RMMResourceMixin, public ::testing::Test {};
 
template <typename... Ts>
class BaseFixtureWithParam : public RMMResourceMixin,
                             public ::testing::TestWithParam<std::tuple<Ts...>> {};
 
using FloatingPointTypes = ::testing::Types<float, double>;
 
}  // namespace test
}  // namespace cuspatial