test_operator_overloading.py revision 12037:d28054ac6ec9
1def 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]"
19
20    v1 += v2
21    v1 *= 2
22    assert str(v1) == "[8.000000, 2.000000]"
23
24    cstats = ConstructorStats.get(Vector2)
25    assert cstats.alive() == 2
26    del v1
27    assert cstats.alive() == 1
28    del v2
29    assert cstats.alive() == 0
30    assert cstats.values() == ['[1.000000, 2.000000]', '[3.000000, -1.000000]',
31                               '[4.000000, 1.000000]', '[-2.000000, 3.000000]',
32                               '[-7.000000, -6.000000]', '[9.000000, 10.000000]',
33                               '[8.000000, 16.000000]', '[0.125000, 0.250000]',
34                               '[7.000000, 6.000000]', '[9.000000, 10.000000]',
35                               '[8.000000, 16.000000]', '[8.000000, 4.000000]']
36    assert cstats.default_constructions == 0
37    assert cstats.copy_constructions == 0
38    assert cstats.move_constructions >= 10
39    assert cstats.copy_assignments == 0
40    assert cstats.move_assignments == 0
41