mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-11 08:03:55 +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.
77 lines
2.8 KiB
C++
77 lines
2.8 KiB
C++
/*
|
|
tests/test_operator_overloading.cpp -- operator overloading
|
|
|
|
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"
|
|
#include "constructor_stats.h"
|
|
#include <pybind11/operators.h>
|
|
|
|
class Vector2 {
|
|
public:
|
|
Vector2(float x, float y) : x(x), y(y) { print_created(this, toString()); }
|
|
Vector2(const Vector2 &v) : x(v.x), y(v.y) { print_copy_created(this); }
|
|
Vector2(Vector2 &&v) : x(v.x), y(v.y) { print_move_created(this); v.x = v.y = 0; }
|
|
~Vector2() { print_destroyed(this); }
|
|
|
|
std::string toString() const {
|
|
return "[" + std::to_string(x) + ", " + std::to_string(y) + "]";
|
|
}
|
|
|
|
void operator=(const Vector2 &v) {
|
|
print_copy_assigned(this);
|
|
x = v.x;
|
|
y = v.y;
|
|
}
|
|
|
|
void operator=(Vector2 &&v) {
|
|
print_move_assigned(this);
|
|
x = v.x; y = v.y; v.x = v.y = 0;
|
|
}
|
|
|
|
Vector2 operator+(const Vector2 &v) const { return Vector2(x + v.x, y + v.y); }
|
|
Vector2 operator-(const Vector2 &v) const { return Vector2(x - v.x, y - v.y); }
|
|
Vector2 operator-(float value) const { return Vector2(x - value, y - value); }
|
|
Vector2 operator+(float value) const { return Vector2(x + value, y + value); }
|
|
Vector2 operator*(float value) const { return Vector2(x * value, y * value); }
|
|
Vector2 operator/(float value) const { return Vector2(x / value, y / value); }
|
|
Vector2& operator+=(const Vector2 &v) { x += v.x; y += v.y; return *this; }
|
|
Vector2& operator-=(const Vector2 &v) { x -= v.x; y -= v.y; return *this; }
|
|
Vector2& operator*=(float v) { x *= v; y *= v; return *this; }
|
|
Vector2& operator/=(float v) { x /= v; y /= v; return *this; }
|
|
|
|
friend Vector2 operator+(float f, const Vector2 &v) { return Vector2(f + v.x, f + v.y); }
|
|
friend Vector2 operator-(float f, const Vector2 &v) { return Vector2(f - v.x, f - v.y); }
|
|
friend Vector2 operator*(float f, const Vector2 &v) { return Vector2(f * v.x, f * v.y); }
|
|
friend Vector2 operator/(float f, const Vector2 &v) { return Vector2(f / v.x, f / v.y); }
|
|
private:
|
|
float x, y;
|
|
};
|
|
|
|
test_initializer operator_overloading([](py::module &m) {
|
|
py::class_<Vector2>(m, "Vector2")
|
|
.def(py::init<float, float>())
|
|
.def(py::self + py::self)
|
|
.def(py::self + float())
|
|
.def(py::self - py::self)
|
|
.def(py::self - float())
|
|
.def(py::self * float())
|
|
.def(py::self / float())
|
|
.def(py::self += py::self)
|
|
.def(py::self -= py::self)
|
|
.def(py::self *= float())
|
|
.def(py::self /= float())
|
|
.def(float() + py::self)
|
|
.def(float() - py::self)
|
|
.def(float() * py::self)
|
|
.def(float() / py::self)
|
|
.def("__str__", &Vector2::toString)
|
|
;
|
|
|
|
m.attr("Vector") = m.attr("Vector2");
|
|
});
|