156 lines
4.3 KiB
Python
156 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
import pytest
|
|
|
|
import env
|
|
from pybind11_tests import ConstructorStats
|
|
from pybind11_tests import operators as m
|
|
|
|
|
|
def test_operator_overloading():
|
|
v1 = m.Vector2(1, 2)
|
|
v2 = m.Vector(3, -1)
|
|
v3 = m.Vector2(1, 2) # Same value as v1, but different instance.
|
|
assert v1 is not v3
|
|
|
|
assert str(v1) == "[1.000000, 2.000000]"
|
|
assert str(v2) == "[3.000000, -1.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]"
|
|
|
|
assert v1 == v3
|
|
assert v1 != v2
|
|
assert hash(v1) == 4
|
|
# TODO(eric.cousineau): Make this work.
|
|
# assert abs(v1) == "abs(Vector2)"
|
|
|
|
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]"
|
|
|
|
cstats = ConstructorStats.get(m.Vector2)
|
|
assert cstats.alive() == 3
|
|
del v1
|
|
assert cstats.alive() == 2
|
|
del v2
|
|
assert cstats.alive() == 1
|
|
del v3
|
|
assert cstats.alive() == 0
|
|
assert cstats.values() == [
|
|
"[1.000000, 2.000000]",
|
|
"[3.000000, -1.000000]",
|
|
"[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 # Shouldn't delete while abase is still alive
|
|
pytest.gc_collect()
|
|
|
|
assert abase.value == 42
|
|
del abase, b
|
|
pytest.gc_collect()
|
|
|
|
|
|
def test_overriding_eq_reset_hash():
|
|
|
|
assert m.Comparable(15) is not m.Comparable(15)
|
|
assert m.Comparable(15) == m.Comparable(15)
|
|
|
|
with pytest.raises(TypeError) as excinfo:
|
|
hash(m.Comparable(15))
|
|
assert str(excinfo.value).startswith("unhashable type:")
|
|
|
|
for hashable in (m.Hashable, m.Hashable2):
|
|
assert hashable(15) is not hashable(15)
|
|
assert hashable(15) == hashable(15)
|
|
|
|
assert hash(hashable(15)) == 15
|
|
assert hash(hashable(15)) == hash(hashable(15))
|
|
|
|
|
|
def test_return_set_of_unhashable():
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_unhashable_HashMe_set()
|
|
if not env.PY2:
|
|
assert str(excinfo.value.__cause__).startswith("unhashable type:")
|