diff options
author | myrtle <gatecat@ds0.me> | 2022-09-15 09:06:35 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-09-15 09:06:35 +0200 |
commit | 3983d4fe53e2c609a5c76510aff8e998a4c22285 (patch) | |
tree | 1c4a543f661dd1b281aecf4660388491702fa8d8 /3rdparty/pybind11/tests/test_sequences_and_iterators.cpp | |
parent | f1349e114f3a16ccd002e8513339e18f5be4d31b (diff) | |
parent | a72f898ff4c4237424c468044a6db9d6953b541e (diff) | |
download | nextpnr-3983d4fe53e2c609a5c76510aff8e998a4c22285.tar.gz nextpnr-3983d4fe53e2c609a5c76510aff8e998a4c22285.tar.bz2 nextpnr-3983d4fe53e2c609a5c76510aff8e998a4c22285.zip |
Merge pull request #1024 from YosysHQ/gatecat/pybind11-bump
3rdparty: Bump vendored pybind11 version for py3.11 support
Diffstat (limited to '3rdparty/pybind11/tests/test_sequences_and_iterators.cpp')
-rw-r--r-- | 3rdparty/pybind11/tests/test_sequences_and_iterators.cpp | 420 |
1 files changed, 312 insertions, 108 deletions
diff --git a/3rdparty/pybind11/tests/test_sequences_and_iterators.cpp b/3rdparty/pybind11/tests/test_sequences_and_iterators.cpp index d318052a..b867f49a 100644 --- a/3rdparty/pybind11/tests/test_sequences_and_iterators.cpp +++ b/3rdparty/pybind11/tests/test_sequences_and_iterators.cpp @@ -8,38 +8,91 @@ BSD-style license that can be found in the LICENSE file. */ -#include "pybind11_tests.h" -#include "constructor_stats.h" #include <pybind11/operators.h> #include <pybind11/stl.h> +#include "constructor_stats.h" +#include "pybind11_tests.h" + #include <algorithm> +#include <utility> +#include <vector> -template<typename T> +#ifdef PYBIND11_HAS_OPTIONAL +# include <optional> +#endif // PYBIND11_HAS_OPTIONAL + +template <typename T> class NonZeroIterator { - const T* ptr_; + const T *ptr_; + public: - NonZeroIterator(const T* ptr) : ptr_(ptr) {} - const T& operator*() const { return *ptr_; } - NonZeroIterator& operator++() { ++ptr_; return *this; } + explicit NonZeroIterator(const T *ptr) : ptr_(ptr) {} + const T &operator*() const { return *ptr_; } + NonZeroIterator &operator++() { + ++ptr_; + return *this; + } }; class NonZeroSentinel {}; -template<typename A, typename B> -bool operator==(const NonZeroIterator<std::pair<A, B>>& it, const NonZeroSentinel&) { +template <typename A, typename B> +bool operator==(const NonZeroIterator<std::pair<A, B>> &it, const NonZeroSentinel &) { return !(*it).first || !(*it).second; } +/* Iterator where dereferencing returns prvalues instead of references. */ +template <typename T> +class NonRefIterator { + const T *ptr_; + +public: + explicit NonRefIterator(const T *ptr) : ptr_(ptr) {} + T operator*() const { return T(*ptr_); } + NonRefIterator &operator++() { + ++ptr_; + return *this; + } + bool operator==(const NonRefIterator &other) const { return ptr_ == other.ptr_; } +}; + +class NonCopyableInt { +public: + explicit NonCopyableInt(int value) : value_(value) {} + NonCopyableInt(const NonCopyableInt &) = delete; + NonCopyableInt(NonCopyableInt &&other) noexcept : value_(other.value_) { + other.value_ = -1; // detect when an unwanted move occurs + } + NonCopyableInt &operator=(const NonCopyableInt &) = delete; + NonCopyableInt &operator=(NonCopyableInt &&other) noexcept { + value_ = other.value_; + other.value_ = -1; // detect when an unwanted move occurs + return *this; + } + int get() const { return value_; } + void set(int value) { value_ = value; } + ~NonCopyableInt() = default; + +private: + int value_; +}; +using NonCopyableIntPair = std::pair<NonCopyableInt, NonCopyableInt>; +PYBIND11_MAKE_OPAQUE(std::vector<NonCopyableInt>); +PYBIND11_MAKE_OPAQUE(std::vector<NonCopyableIntPair>); + template <typename PythonType> py::list test_random_access_iterator(PythonType x) { - if (x.size() < 5) + if (x.size() < 5) { throw py::value_error("Please provide at least 5 elements for testing."); + } auto checks = py::list(); auto assert_equal = [&checks](py::handle a, py::handle b) { auto result = PyObject_RichCompareBool(a.ptr(), b.ptr(), Py_EQ); - if (result == -1) { throw py::error_already_set(); } + if (result == -1) { + throw py::error_already_set(); + } checks.append(result != 0); }; @@ -74,63 +127,83 @@ py::list test_random_access_iterator(PythonType x) { TEST_SUBMODULE(sequences_and_iterators, m) { // test_sliceable - class Sliceable{ + class Sliceable { public: - Sliceable(int n): size(n) {} - int start,stop,step; - int size; + explicit Sliceable(int n) : size(n) {} + int start, stop, step; + int size; }; - py::class_<Sliceable>(m,"Sliceable") + py::class_<Sliceable>(m, "Sliceable") .def(py::init<int>()) - .def("__getitem__",[](const Sliceable &s, py::slice slice) { - py::ssize_t start, stop, step, slicelength; - if (!slice.compute(s.size, &start, &stop, &step, &slicelength)) - throw py::error_already_set(); - int istart = static_cast<int>(start); - int istop = static_cast<int>(stop); - int istep = static_cast<int>(step); - return std::make_tuple(istart,istop,istep); - }) - ; + .def("__getitem__", [](const Sliceable &s, const py::slice &slice) { + py::ssize_t start = 0, stop = 0, step = 0, slicelength = 0; + if (!slice.compute(s.size, &start, &stop, &step, &slicelength)) { + throw py::error_already_set(); + } + int istart = static_cast<int>(start); + int istop = static_cast<int>(stop); + int istep = static_cast<int>(step); + return std::make_tuple(istart, istop, istep); + }); + + m.def("make_forward_slice_size_t", []() { return py::slice(0, -1, 1); }); + m.def("make_reversed_slice_object", + []() { return py::slice(py::none(), py::none(), py::int_(-1)); }); +#ifdef PYBIND11_HAS_OPTIONAL + m.attr("has_optional") = true; + m.def("make_reversed_slice_size_t_optional_verbose", + []() { return py::slice(std::nullopt, std::nullopt, -1); }); + // Warning: The following spelling may still compile if optional<> is not present and give + // wrong answers. Please use with caution. + m.def("make_reversed_slice_size_t_optional", []() { return py::slice({}, {}, -1); }); +#else + m.attr("has_optional") = false; +#endif // test_sequence class Sequence { public: - Sequence(size_t size) : m_size(size) { + explicit Sequence(size_t size) : m_size(size) { print_created(this, "of size", m_size); + // NOLINTNEXTLINE(cppcoreguidelines-prefer-member-initializer) m_data = new float[size]; memset(m_data, 0, sizeof(float) * size); } - Sequence(const std::vector<float> &value) : m_size(value.size()) { + explicit Sequence(const std::vector<float> &value) : m_size(value.size()) { print_created(this, "of size", m_size, "from std::vector"); + // NOLINTNEXTLINE(cppcoreguidelines-prefer-member-initializer) m_data = new float[m_size]; memcpy(m_data, &value[0], sizeof(float) * m_size); } Sequence(const Sequence &s) : m_size(s.m_size) { print_copy_created(this); + // NOLINTNEXTLINE(cppcoreguidelines-prefer-member-initializer) m_data = new float[m_size]; - memcpy(m_data, s.m_data, sizeof(float)*m_size); + memcpy(m_data, s.m_data, sizeof(float) * m_size); } - Sequence(Sequence &&s) : m_size(s.m_size), m_data(s.m_data) { + Sequence(Sequence &&s) noexcept : m_size(s.m_size), m_data(s.m_data) { print_move_created(this); s.m_size = 0; s.m_data = nullptr; } - ~Sequence() { print_destroyed(this); delete[] m_data; } + ~Sequence() { + print_destroyed(this); + delete[] m_data; + } Sequence &operator=(const Sequence &s) { if (&s != this) { delete[] m_data; m_size = s.m_size; m_data = new float[m_size]; - memcpy(m_data, s.m_data, sizeof(float)*m_size); + memcpy(m_data, s.m_data, sizeof(float) * m_size); } print_copy_assigned(this); return *this; } - Sequence &operator=(Sequence &&s) { + Sequence &operator=(Sequence &&s) noexcept { if (&s != this) { delete[] m_data; m_size = s.m_size; @@ -143,10 +216,14 @@ TEST_SUBMODULE(sequences_and_iterators, m) { } bool operator==(const Sequence &s) const { - if (m_size != s.size()) return false; - for (size_t i = 0; i < m_size; ++i) - if (m_data[i] != s[i]) + if (m_size != s.size()) { + return false; + } + for (size_t i = 0; i < m_size; ++i) { + if (m_data[i] != s[i]) { return false; + } + } return true; } bool operator!=(const Sequence &s) const { return !operator==(s); } @@ -155,23 +232,26 @@ TEST_SUBMODULE(sequences_and_iterators, m) { float &operator[](size_t index) { return m_data[index]; } bool contains(float v) const { - for (size_t i = 0; i < m_size; ++i) - if (v == m_data[i]) + for (size_t i = 0; i < m_size; ++i) { + if (v == m_data[i]) { return true; + } + } return false; } Sequence reversed() const { Sequence result(m_size); - for (size_t i = 0; i < m_size; ++i) + for (size_t i = 0; i < m_size; ++i) { result[m_size - i - 1] = m_data[i]; + } return result; } size_t size() const { return m_size; } const float *begin() const { return m_data; } - const float *end() const { return m_data+m_size; } + const float *end() const { return m_data + m_size; } private: size_t m_size; @@ -179,43 +259,59 @@ TEST_SUBMODULE(sequences_and_iterators, m) { }; py::class_<Sequence>(m, "Sequence") .def(py::init<size_t>()) - .def(py::init<const std::vector<float>&>()) + .def(py::init<const std::vector<float> &>()) /// Bare bones interface - .def("__getitem__", [](const Sequence &s, size_t i) { - if (i >= s.size()) throw py::index_error(); - return s[i]; - }) - .def("__setitem__", [](Sequence &s, size_t i, float v) { - if (i >= s.size()) throw py::index_error(); - s[i] = v; - }) + .def("__getitem__", + [](const Sequence &s, size_t i) { + if (i >= s.size()) { + throw py::index_error(); + } + return s[i]; + }) + .def("__setitem__", + [](Sequence &s, size_t i, float v) { + if (i >= s.size()) { + throw py::index_error(); + } + s[i] = v; + }) .def("__len__", &Sequence::size) /// Optional sequence protocol operations - .def("__iter__", [](const Sequence &s) { return py::make_iterator(s.begin(), s.end()); }, - py::keep_alive<0, 1>() /* Essential: keep object alive while iterator exists */) + .def( + "__iter__", + [](const Sequence &s) { return py::make_iterator(s.begin(), s.end()); }, + py::keep_alive<0, 1>() /* Essential: keep object alive while iterator exists */) .def("__contains__", [](const Sequence &s, float v) { return s.contains(v); }) .def("__reversed__", [](const Sequence &s) -> Sequence { return s.reversed(); }) /// Slicing protocol (optional) - .def("__getitem__", [](const Sequence &s, py::slice slice) -> Sequence* { - size_t start, stop, step, slicelength; - if (!slice.compute(s.size(), &start, &stop, &step, &slicelength)) - throw py::error_already_set(); - auto *seq = new Sequence(slicelength); - for (size_t i = 0; i < slicelength; ++i) { - (*seq)[i] = s[start]; start += step; - } - return seq; - }) - .def("__setitem__", [](Sequence &s, py::slice slice, const Sequence &value) { - size_t start, stop, step, slicelength; - if (!slice.compute(s.size(), &start, &stop, &step, &slicelength)) - throw py::error_already_set(); - if (slicelength != value.size()) - throw std::runtime_error("Left and right hand size of slice assignment have different sizes!"); - for (size_t i = 0; i < slicelength; ++i) { - s[start] = value[i]; start += step; - } - }) + .def("__getitem__", + [](const Sequence &s, const py::slice &slice) -> Sequence * { + size_t start = 0, stop = 0, step = 0, slicelength = 0; + if (!slice.compute(s.size(), &start, &stop, &step, &slicelength)) { + throw py::error_already_set(); + } + auto *seq = new Sequence(slicelength); + for (size_t i = 0; i < slicelength; ++i) { + (*seq)[i] = s[start]; + start += step; + } + return seq; + }) + .def("__setitem__", + [](Sequence &s, const py::slice &slice, const Sequence &value) { + size_t start = 0, stop = 0, step = 0, slicelength = 0; + if (!slice.compute(s.size(), &start, &stop, &step, &slicelength)) { + throw py::error_already_set(); + } + if (slicelength != value.size()) { + throw std::runtime_error( + "Left and right hand size of slice assignment have different sizes!"); + } + for (size_t i = 0; i < slicelength; ++i) { + s[start] = value[i]; + start += step; + } + }) /// Comparisons .def(py::self == py::self) .def(py::self != py::self) @@ -223,19 +319,21 @@ TEST_SUBMODULE(sequences_and_iterators, m) { ; // test_map_iterator - // Interface of a map-like object that isn't (directly) an unordered_map, but provides some basic - // map-like functionality. + // Interface of a map-like object that isn't (directly) an unordered_map, but provides some + // basic map-like functionality. class StringMap { public: StringMap() = default; - StringMap(std::unordered_map<std::string, std::string> init) + explicit StringMap(std::unordered_map<std::string, std::string> init) : map(std::move(init)) {} - void set(std::string key, std::string val) { map[key] = val; } - std::string get(std::string key) const { return map.at(key); } + void set(const std::string &key, std::string val) { map[key] = std::move(val); } + std::string get(const std::string &key) const { return map.at(key); } size_t size() const { return map.size(); } + private: std::unordered_map<std::string, std::string> map; + public: decltype(map.cbegin()) begin() const { return map.cbegin(); } decltype(map.cend()) end() const { return map.cend(); } @@ -243,38 +341,142 @@ TEST_SUBMODULE(sequences_and_iterators, m) { py::class_<StringMap>(m, "StringMap") .def(py::init<>()) .def(py::init<std::unordered_map<std::string, std::string>>()) - .def("__getitem__", [](const StringMap &map, std::string key) { - try { return map.get(key); } - catch (const std::out_of_range&) { - throw py::key_error("key '" + key + "' does not exist"); - } - }) + .def("__getitem__", + [](const StringMap &map, const std::string &key) { + try { + return map.get(key); + } catch (const std::out_of_range &) { + throw py::key_error("key '" + key + "' does not exist"); + } + }) .def("__setitem__", &StringMap::set) .def("__len__", &StringMap::size) - .def("__iter__", [](const StringMap &map) { return py::make_key_iterator(map.begin(), map.end()); }, - py::keep_alive<0, 1>()) - .def("items", [](const StringMap &map) { return py::make_iterator(map.begin(), map.end()); }, - py::keep_alive<0, 1>()) - ; + .def( + "__iter__", + [](const StringMap &map) { return py::make_key_iterator(map.begin(), map.end()); }, + py::keep_alive<0, 1>()) + .def( + "items", + [](const StringMap &map) { return py::make_iterator(map.begin(), map.end()); }, + py::keep_alive<0, 1>()) + .def( + "values", + [](const StringMap &map) { return py::make_value_iterator(map.begin(), map.end()); }, + py::keep_alive<0, 1>()); // test_generalized_iterators class IntPairs { public: - IntPairs(std::vector<std::pair<int, int>> data) : data_(std::move(data)) {} - const std::pair<int, int>* begin() const { return data_.data(); } + explicit IntPairs(std::vector<std::pair<int, int>> data) : data_(std::move(data)) {} + const std::pair<int, int> *begin() const { return data_.data(); } + // .end() only required for py::make_iterator(self) overload + const std::pair<int, int> *end() const { return data_.data() + data_.size(); } + private: std::vector<std::pair<int, int>> data_; }; py::class_<IntPairs>(m, "IntPairs") .def(py::init<std::vector<std::pair<int, int>>>()) - .def("nonzero", [](const IntPairs& s) { - return py::make_iterator(NonZeroIterator<std::pair<int, int>>(s.begin()), NonZeroSentinel()); - }, py::keep_alive<0, 1>()) - .def("nonzero_keys", [](const IntPairs& s) { - return py::make_key_iterator(NonZeroIterator<std::pair<int, int>>(s.begin()), NonZeroSentinel()); - }, py::keep_alive<0, 1>()) - ; - + .def( + "nonzero", + [](const IntPairs &s) { + return py::make_iterator(NonZeroIterator<std::pair<int, int>>(s.begin()), + NonZeroSentinel()); + }, + py::keep_alive<0, 1>()) + .def( + "nonzero_keys", + [](const IntPairs &s) { + return py::make_key_iterator(NonZeroIterator<std::pair<int, int>>(s.begin()), + NonZeroSentinel()); + }, + py::keep_alive<0, 1>()) + .def( + "nonzero_values", + [](const IntPairs &s) { + return py::make_value_iterator(NonZeroIterator<std::pair<int, int>>(s.begin()), + NonZeroSentinel()); + }, + py::keep_alive<0, 1>()) + + // test iterator that returns values instead of references + .def( + "nonref", + [](const IntPairs &s) { + return py::make_iterator(NonRefIterator<std::pair<int, int>>(s.begin()), + NonRefIterator<std::pair<int, int>>(s.end())); + }, + py::keep_alive<0, 1>()) + .def( + "nonref_keys", + [](const IntPairs &s) { + return py::make_key_iterator(NonRefIterator<std::pair<int, int>>(s.begin()), + NonRefIterator<std::pair<int, int>>(s.end())); + }, + py::keep_alive<0, 1>()) + .def( + "nonref_values", + [](const IntPairs &s) { + return py::make_value_iterator(NonRefIterator<std::pair<int, int>>(s.begin()), + NonRefIterator<std::pair<int, int>>(s.end())); + }, + py::keep_alive<0, 1>()) + + // test single-argument make_iterator + .def( + "simple_iterator", + [](IntPairs &self) { return py::make_iterator(self); }, + py::keep_alive<0, 1>()) + .def( + "simple_keys", + [](IntPairs &self) { return py::make_key_iterator(self); }, + py::keep_alive<0, 1>()) + .def( + "simple_values", + [](IntPairs &self) { return py::make_value_iterator(self); }, + py::keep_alive<0, 1>()) + + // Test iterator with an Extra (doesn't do anything useful, so not used + // at runtime, but tests need to be able to compile with the correct + // overload. See PR #3293. + .def( + "_make_iterator_extras", + [](IntPairs &self) { return py::make_iterator(self, py::call_guard<int>()); }, + py::keep_alive<0, 1>()) + .def( + "_make_key_extras", + [](IntPairs &self) { return py::make_key_iterator(self, py::call_guard<int>()); }, + py::keep_alive<0, 1>()) + .def( + "_make_value_extras", + [](IntPairs &self) { return py::make_value_iterator(self, py::call_guard<int>()); }, + py::keep_alive<0, 1>()); + + // test_iterator_referencing + py::class_<NonCopyableInt>(m, "NonCopyableInt") + .def(py::init<int>()) + .def("set", &NonCopyableInt::set) + .def("__int__", &NonCopyableInt::get); + py::class_<std::vector<NonCopyableInt>>(m, "VectorNonCopyableInt") + .def(py::init<>()) + .def("append", + [](std::vector<NonCopyableInt> &vec, int value) { vec.emplace_back(value); }) + .def("__iter__", [](std::vector<NonCopyableInt> &vec) { + return py::make_iterator(vec.begin(), vec.end()); + }); + py::class_<std::vector<NonCopyableIntPair>>(m, "VectorNonCopyableIntPair") + .def(py::init<>()) + .def("append", + [](std::vector<NonCopyableIntPair> &vec, const std::pair<int, int> &value) { + vec.emplace_back(NonCopyableInt(value.first), NonCopyableInt(value.second)); + }) + .def("keys", + [](std::vector<NonCopyableIntPair> &vec) { + return py::make_key_iterator(vec.begin(), vec.end()); + }) + .def("values", [](std::vector<NonCopyableIntPair> &vec) { + return py::make_value_iterator(vec.begin(), vec.end()); + }); #if 0 // Obsolete: special data structure for exposing custom iterator types to python @@ -304,7 +506,7 @@ TEST_SUBMODULE(sequences_and_iterators, m) { #endif // test_python_iterator_in_cpp - m.def("object_to_list", [](py::object o) { + m.def("object_to_list", [](const py::object &o) { auto l = py::list(); for (auto item : o) { l.append(item); @@ -322,22 +524,22 @@ TEST_SUBMODULE(sequences_and_iterators, m) { }); // test_sequence_length: check that Python sequences can be converted to py::sequence. - m.def("sequence_length", [](py::sequence seq) { return seq.size(); }); + m.def("sequence_length", [](const py::sequence &seq) { return seq.size(); }); // Make sure that py::iterator works with std algorithms - m.def("count_none", [](py::object o) { + m.def("count_none", [](const py::object &o) { return std::count_if(o.begin(), o.end(), [](py::handle h) { return h.is_none(); }); }); - m.def("find_none", [](py::object o) { + m.def("find_none", [](const py::object &o) { auto it = std::find_if(o.begin(), o.end(), [](py::handle h) { return h.is_none(); }); return it->is_none(); }); - m.def("count_nonzeros", [](py::dict d) { - return std::count_if(d.begin(), d.end(), [](std::pair<py::handle, py::handle> p) { - return p.second.cast<int>() != 0; - }); + m.def("count_nonzeros", [](const py::dict &d) { + return std::count_if(d.begin(), d.end(), [](std::pair<py::handle, py::handle> p) { + return p.second.cast<int>() != 0; + }); }); m.def("tuple_iterator", &test_random_access_iterator<py::tuple>); @@ -352,7 +554,9 @@ TEST_SUBMODULE(sequences_and_iterators, m) { // test_iterator_rvp // #388: Can't make iterators via make_iterator() with different r/v policies - static std::vector<int> list = { 1, 2, 3 }; - m.def("make_iterator_1", []() { return py::make_iterator<py::return_value_policy::copy>(list); }); - m.def("make_iterator_2", []() { return py::make_iterator<py::return_value_policy::automatic>(list); }); + static std::vector<int> list = {1, 2, 3}; + m.def("make_iterator_1", + []() { return py::make_iterator<py::return_value_policy::copy>(list); }); + m.def("make_iterator_2", + []() { return py::make_iterator<py::return_value_policy::automatic>(list); }); } |