mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-22 13:15:12 +00:00
52f4be8946
Adding or removing tests is a little bit cumbersome currently: the test needs to be added to CMakeLists.txt, the init function needs to be predeclared in pybind11_tests.cpp, then called in the plugin initialization. While this isn't a big deal for tests that are being committed, it's more of a hassle when working on some new feature or test code for which I temporarily only care about building and linking the test being worked on rather than the entire test suite. This commit changes tests to self-register their initialization by having each test initialize a local object (which stores the initialization function in a static variable). This makes changing the set of tests being build easy: one only needs to add or comment out test names in tests/CMakeLists.txt. A couple other minor changes that go along with this: - test_eigen.cpp is now included in the test list, then removed if eigen isn't available. This lets you disable the eigen tests by commenting it out, just like all the other tests, but keeps the build working without eigen eigen isn't available. (Also, if it's commented out, we don't even bother looking for and reporting the building with/without eigen status message). - pytest is now invoked with all the built test names (with .cpp changed to .py) so that it doesn't try to run tests that weren't built.
52 lines
1.8 KiB
C++
52 lines
1.8 KiB
C++
/*
|
|
tests/test_pickling.cpp -- pickle support
|
|
|
|
Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
|
|
|
|
All rights reserved. Use of this source code is governed by a
|
|
BSD-style license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
#include "pybind11_tests.h"
|
|
|
|
class Pickleable {
|
|
public:
|
|
Pickleable(const std::string &value) : m_value(value) { }
|
|
const std::string &value() const { return m_value; }
|
|
|
|
void setExtra1(int extra1) { m_extra1 = extra1; }
|
|
void setExtra2(int extra2) { m_extra2 = extra2; }
|
|
int extra1() const { return m_extra1; }
|
|
int extra2() const { return m_extra2; }
|
|
private:
|
|
std::string m_value;
|
|
int m_extra1 = 0;
|
|
int m_extra2 = 0;
|
|
};
|
|
|
|
test_initializer pickling([](py::module &m) {
|
|
py::class_<Pickleable>(m, "Pickleable")
|
|
.def(py::init<std::string>())
|
|
.def("value", &Pickleable::value)
|
|
.def("extra1", &Pickleable::extra1)
|
|
.def("extra2", &Pickleable::extra2)
|
|
.def("setExtra1", &Pickleable::setExtra1)
|
|
.def("setExtra2", &Pickleable::setExtra2)
|
|
// For details on the methods below, refer to
|
|
// http://docs.python.org/3/library/pickle.html#pickling-class-instances
|
|
.def("__getstate__", [](const Pickleable &p) {
|
|
/* Return a tuple that fully encodes the state of the object */
|
|
return py::make_tuple(p.value(), p.extra1(), p.extra2());
|
|
})
|
|
.def("__setstate__", [](Pickleable &p, py::tuple t) {
|
|
if (t.size() != 3)
|
|
throw std::runtime_error("Invalid state!");
|
|
/* Invoke the constructor (need to use in-place version) */
|
|
new (&p) Pickleable(t[0].cast<std::string>());
|
|
|
|
/* Assign any additional state */
|
|
p.setExtra1(t[1].cast<int>());
|
|
p.setExtra2(t[2].cast<int>());
|
|
});
|
|
});
|