1import pytest
2from pybind11_tests import operators as m
3from pybind11_tests import ConstructorStats
4
5
6def test_operator_overloading():
7    v1 = m.Vector2(1, 2)
8    v2 = m.Vector(3, -1)
9    assert str(v1) == "[1.000000, 2.000000]"
10    assert str(v2) == "[3.000000, -1.000000]"
11
12    assert str(-v2) == "[-3.000000, 1.000000]"
13
14    assert str(v1 + v2) == "[4.000000, 1.000000]"
15    assert str(v1 - v2) == "[-2.000000, 3.000000]"
16    assert str(v1 - 8) == "[-7.000000, -6.000000]"
17    assert str(v1 + 8) == "[9.000000, 10.000000]"
18    assert str(v1 * 8) == "[8.000000, 16.000000]"
19    assert str(v1 / 8) == "[0.125000, 0.250000]"
20    assert str(8 - v1) == "[7.000000, 6.000000]"
21    assert str(8 + v1) == "[9.000000, 10.000000]"
22    assert str(8 * v1) == "[8.000000, 16.000000]"
23    assert str(8 / v1) == "[8.000000, 4.000000]"
24    assert str(v1 * v2) == "[3.000000, -2.000000]"
25    assert str(v2 / v1) == "[3.000000, -0.500000]"
26
27    v1 += 2 * v2
28    assert str(v1) == "[7.000000, 0.000000]"
29    v1 -= v2
30    assert str(v1) == "[4.000000, 1.000000]"
31    v1 *= 2
32    assert str(v1) == "[8.000000, 2.000000]"
33    v1 /= 16
34    assert str(v1) == "[0.500000, 0.125000]"
35    v1 *= v2
36    assert str(v1) == "[1.500000, -0.125000]"
37    v2 /= v1
38    assert str(v2) == "[2.000000, 8.000000]"
39
40    assert hash(v1) == 4
41
42    cstats = ConstructorStats.get(m.Vector2)
43    assert cstats.alive() == 2
44    del v1
45    assert cstats.alive() == 1
46    del v2
47    assert cstats.alive() == 0
48    assert cstats.values() == ['[1.000000, 2.000000]', '[3.000000, -1.000000]',
49                               '[-3.000000, 1.000000]', '[4.000000, 1.000000]',
50                               '[-2.000000, 3.000000]', '[-7.000000, -6.000000]',
51                               '[9.000000, 10.000000]', '[8.000000, 16.000000]',
52                               '[0.125000, 0.250000]', '[7.000000, 6.000000]',
53                               '[9.000000, 10.000000]', '[8.000000, 16.000000]',
54                               '[8.000000, 4.000000]', '[3.000000, -2.000000]',
55                               '[3.000000, -0.500000]', '[6.000000, -2.000000]']
56    assert cstats.default_constructions == 0
57    assert cstats.copy_constructions == 0
58    assert cstats.move_constructions >= 10
59    assert cstats.copy_assignments == 0
60    assert cstats.move_assignments == 0
61
62
63def test_operators_notimplemented():
64    """#393: need to return NotSupported to ensure correct arithmetic operator behavior"""
65
66    c1, c2 = m.C1(), m.C2()
67    assert c1 + c1 == 11
68    assert c2 + c2 == 22
69    assert c2 + c1 == 21
70    assert c1 + c2 == 12
71
72
73def test_nested():
74    """#328: first member in a class can't be used in operators"""
75
76    a = m.NestA()
77    b = m.NestB()
78    c = m.NestC()
79
80    a += 10
81    assert m.get_NestA(a) == 13
82    b.a += 100
83    assert m.get_NestA(b.a) == 103
84    c.b.a += 1000
85    assert m.get_NestA(c.b.a) == 1003
86    b -= 1
87    assert m.get_NestB(b) == 3
88    c.b -= 3
89    assert m.get_NestB(c.b) == 1
90    c *= 7
91    assert m.get_NestC(c) == 35
92
93    abase = a.as_base()
94    assert abase.value == -2
95    a.as_base().value += 44
96    assert abase.value == 42
97    assert c.b.a.as_base().value == -2
98    c.b.a.as_base().value += 44
99    assert c.b.a.as_base().value == 42
100
101    del c
102    pytest.gc_collect()
103    del a  # Shouldn't delete while abase is still alive
104    pytest.gc_collect()
105
106    assert abase.value == 42
107    del abase, b
108    pytest.gc_collect()
109