aboutsummaryrefslogtreecommitdiff
path: root/tests/test_vector_unique_ptr_member.cpp
blob: 680cf9a6d8c9fad7b0d7af0aee0dabfdc68e4c12 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include "pybind11_tests.h"

#include <cstddef>
#include <memory>
#include <vector>

namespace pybind11_tests {
namespace vector_unique_ptr_member {

struct DataType {};

// Reduced from a use case in the wild.
struct VectorOwner {
    static std::unique_ptr<VectorOwner> Create(std::size_t num_elems) {
        return std::unique_ptr<VectorOwner>(
            new VectorOwner(std::vector<std::unique_ptr<DataType>>(num_elems)));
    }

    std::size_t data_size() const { return data_.size(); }

private:
    explicit VectorOwner(std::vector<std::unique_ptr<DataType>> data) : data_(std::move(data)) {}

    const std::vector<std::unique_ptr<DataType>> data_;
};

} // namespace vector_unique_ptr_member
} // namespace pybind11_tests

namespace pybind11 {
namespace detail {

template <>
struct is_copy_constructible<pybind11_tests::vector_unique_ptr_member::VectorOwner>
    : std::false_type {};

template <>
struct is_move_constructible<pybind11_tests::vector_unique_ptr_member::VectorOwner>
    : std::false_type {};

} // namespace detail
} // namespace pybind11

using namespace pybind11_tests::vector_unique_ptr_member;

py::object py_cast_VectorOwner_ptr(VectorOwner *ptr) { return py::cast(ptr); }

TEST_SUBMODULE(vector_unique_ptr_member, m) {
    py::class_<VectorOwner>(m, "VectorOwner")
        .def_static("Create", &VectorOwner::Create)
        .def("data_size", &VectorOwner::data_size);

    m.def("py_cast_VectorOwner_ptr", py_cast_VectorOwner_ptr);
}