| import pytest |
| from pybind11_tests import operators as m |
| from pybind11_tests import ConstructorStats |
| |
| |
| def test_operator_overloading(): |
| v1 = m.Vector2(1, 2) |
| v2 = m.Vector(3, -1) |
| assert str(v1) == "[1.000000, 2.000000]" |
| assert str(v2) == "[3.000000, -1.000000]" |
| |
| assert str(v1 + v2) == "[4.000000, 1.000000]" |
| assert str(v1 - v2) == "[-2.000000, 3.000000]" |
| assert str(v1 - 8) == "[-7.000000, -6.000000]" |
| assert str(v1 + 8) == "[9.000000, 10.000000]" |
| assert str(v1 * 8) == "[8.000000, 16.000000]" |
| assert str(v1 / 8) == "[0.125000, 0.250000]" |
| assert str(8 - v1) == "[7.000000, 6.000000]" |
| assert str(8 + v1) == "[9.000000, 10.000000]" |
| assert str(8 * v1) == "[8.000000, 16.000000]" |
| assert str(8 / v1) == "[8.000000, 4.000000]" |
| assert str(v1 * v2) == "[3.000000, -2.000000]" |
| assert str(v2 / v1) == "[3.000000, -0.500000]" |
| |
| v1 += 2 * v2 |
| assert str(v1) == "[7.000000, 0.000000]" |
| v1 -= v2 |
| assert str(v1) == "[4.000000, 1.000000]" |
| v1 *= 2 |
| assert str(v1) == "[8.000000, 2.000000]" |
| v1 /= 16 |
| assert str(v1) == "[0.500000, 0.125000]" |
| v1 *= v2 |
| assert str(v1) == "[1.500000, -0.125000]" |
| v2 /= v1 |
| assert str(v2) == "[2.000000, 8.000000]" |
| |
| assert hash(v1) == 4 |
| |
| cstats = ConstructorStats.get(m.Vector2) |
| assert cstats.alive() == 2 |
| del v1 |
| assert cstats.alive() == 1 |
| del v2 |
| assert cstats.alive() == 0 |
| assert cstats.values() == ['[1.000000, 2.000000]', '[3.000000, -1.000000]', |
| '[4.000000, 1.000000]', '[-2.000000, 3.000000]', |
| '[-7.000000, -6.000000]', '[9.000000, 10.000000]', |
| '[8.000000, 16.000000]', '[0.125000, 0.250000]', |
| '[7.000000, 6.000000]', '[9.000000, 10.000000]', |
| '[8.000000, 16.000000]', '[8.000000, 4.000000]', |
| '[3.000000, -2.000000]', '[3.000000, -0.500000]', |
| '[6.000000, -2.000000]'] |
| assert cstats.default_constructions == 0 |
| assert cstats.copy_constructions == 0 |
| assert cstats.move_constructions >= 10 |
| assert cstats.copy_assignments == 0 |
| assert cstats.move_assignments == 0 |
| |
| |
| def test_operators_notimplemented(): |
| """#393: need to return NotSupported to ensure correct arithmetic operator behavior""" |
| |
| c1, c2 = m.C1(), m.C2() |
| assert c1 + c1 == 11 |
| assert c2 + c2 == 22 |
| assert c2 + c1 == 21 |
| assert c1 + c2 == 12 |
| |
| |
| def test_nested(): |
| """#328: first member in a class can't be used in operators""" |
| |
| a = m.NestA() |
| b = m.NestB() |
| c = m.NestC() |
| |
| a += 10 |
| assert m.get_NestA(a) == 13 |
| b.a += 100 |
| assert m.get_NestA(b.a) == 103 |
| c.b.a += 1000 |
| assert m.get_NestA(c.b.a) == 1003 |
| b -= 1 |
| assert m.get_NestB(b) == 3 |
| c.b -= 3 |
| assert m.get_NestB(c.b) == 1 |
| c *= 7 |
| assert m.get_NestC(c) == 35 |
| |
| abase = a.as_base() |
| assert abase.value == -2 |
| a.as_base().value += 44 |
| assert abase.value == 42 |
| assert c.b.a.as_base().value == -2 |
| c.b.a.as_base().value += 44 |
| assert c.b.a.as_base().value == 42 |
| |
| del c |
| pytest.gc_collect() |
| del a # Should't delete while abase is still alive |
| pytest.gc_collect() |
| |
| assert abase.value == 42 |
| del abase, b |
| pytest.gc_collect() |