Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 1 | def 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 Rhinelander | acad05c | 2017-05-20 20:19:26 -0400 | [diff] [blame] | 19 | assert str(v1 * v2) == "[3.000000, -2.000000]" |
| 20 | assert str(v2 / v1) == "[3.000000, -0.500000]" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 21 | |
Jason Rhinelander | acad05c | 2017-05-20 20:19:26 -0400 | [diff] [blame] | 22 | v1 += 2 * v2 |
| 23 | assert str(v1) == "[7.000000, 0.000000]" |
| 24 | v1 -= v2 |
| 25 | assert str(v1) == "[4.000000, 1.000000]" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 26 | v1 *= 2 |
| 27 | assert str(v1) == "[8.000000, 2.000000]" |
Jason Rhinelander | acad05c | 2017-05-20 20:19:26 -0400 | [diff] [blame] | 28 | 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 Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 34 | |
| 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 Rhinelander | acad05c | 2017-05-20 20:19:26 -0400 | [diff] [blame] | 46 | '[8.000000, 16.000000]', '[8.000000, 4.000000]', |
| 47 | '[3.000000, -2.000000]', '[3.000000, -0.500000]', |
| 48 | '[6.000000, -2.000000]'] |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 49 | 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 |