blob: dd37c3497a30c5cfa2d6273426be5d0d1cf3d89e [file] [log] [blame]
Dean Moldovana0c1ccf2016-08-12 13:50:00 +02001def test_operator_overloading():
2 from pybind11_tests import Vector2, Vector, ConstructorStats
3
4 v1 = Vector2(1, 2)
5 v2 = Vector(3, -1)
6 assert str(v1) == "[1.000000, 2.000000]"
7 assert str(v2) == "[3.000000, -1.000000]"
8
9 assert str(v1 + v2) == "[4.000000, 1.000000]"
10 assert str(v1 - v2) == "[-2.000000, 3.000000]"
11 assert str(v1 - 8) == "[-7.000000, -6.000000]"
12 assert str(v1 + 8) == "[9.000000, 10.000000]"
13 assert str(v1 * 8) == "[8.000000, 16.000000]"
14 assert str(v1 / 8) == "[0.125000, 0.250000]"
15 assert str(8 - v1) == "[7.000000, 6.000000]"
16 assert str(8 + v1) == "[9.000000, 10.000000]"
17 assert str(8 * v1) == "[8.000000, 16.000000]"
18 assert str(8 / v1) == "[8.000000, 4.000000]"
Jason Rhinelanderacad05c2017-05-20 20:19:26 -040019 assert str(v1 * v2) == "[3.000000, -2.000000]"
20 assert str(v2 / v1) == "[3.000000, -0.500000]"
Dean Moldovana0c1ccf2016-08-12 13:50:00 +020021
Jason Rhinelanderacad05c2017-05-20 20:19:26 -040022 v1 += 2 * v2
23 assert str(v1) == "[7.000000, 0.000000]"
24 v1 -= v2
25 assert str(v1) == "[4.000000, 1.000000]"
Dean Moldovana0c1ccf2016-08-12 13:50:00 +020026 v1 *= 2
27 assert str(v1) == "[8.000000, 2.000000]"
Jason Rhinelanderacad05c2017-05-20 20:19:26 -040028 v1 /= 16
29 assert str(v1) == "[0.500000, 0.125000]"
30 v1 *= v2
31 assert str(v1) == "[1.500000, -0.125000]"
32 v2 /= v1
33 assert str(v2) == "[2.000000, 8.000000]"
Dean Moldovana0c1ccf2016-08-12 13:50:00 +020034
35 cstats = ConstructorStats.get(Vector2)
36 assert cstats.alive() == 2
37 del v1
38 assert cstats.alive() == 1
39 del v2
40 assert cstats.alive() == 0
41 assert cstats.values() == ['[1.000000, 2.000000]', '[3.000000, -1.000000]',
42 '[4.000000, 1.000000]', '[-2.000000, 3.000000]',
43 '[-7.000000, -6.000000]', '[9.000000, 10.000000]',
44 '[8.000000, 16.000000]', '[0.125000, 0.250000]',
45 '[7.000000, 6.000000]', '[9.000000, 10.000000]',
Jason Rhinelanderacad05c2017-05-20 20:19:26 -040046 '[8.000000, 16.000000]', '[8.000000, 4.000000]',
47 '[3.000000, -2.000000]', '[3.000000, -0.500000]',
48 '[6.000000, -2.000000]']
Dean Moldovana0c1ccf2016-08-12 13:50:00 +020049 assert cstats.default_constructions == 0
50 assert cstats.copy_constructions == 0
51 assert cstats.move_constructions >= 10
52 assert cstats.copy_assignments == 0
53 assert cstats.move_assignments == 0