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