mirror of
https://github.com/RYDE-WORK/pybind11.git
synced 2026-01-19 21:23:26 +08:00
e6fd2cd5ab
Enumerations on Python 2.7 were not always implicitly converted to integers (depending on the target size). This patch adds a __long__ conversion function (only enabled on 2.7) which fixes this issue. The attached test case fails without this patch.
129 lines
4.0 KiB
Python
129 lines
4.0 KiB
Python
import pytest
|
|
|
|
|
|
def test_unscoped_enum():
|
|
from pybind11_tests import UnscopedEnum, EOne
|
|
|
|
assert str(UnscopedEnum.EOne) == "UnscopedEnum.EOne"
|
|
assert str(UnscopedEnum.ETwo) == "UnscopedEnum.ETwo"
|
|
assert str(EOne) == "UnscopedEnum.EOne"
|
|
# __members__ property
|
|
assert UnscopedEnum.__members__ == {"EOne": UnscopedEnum.EOne, "ETwo": UnscopedEnum.ETwo}
|
|
# __members__ readonly
|
|
with pytest.raises(AttributeError):
|
|
UnscopedEnum.__members__ = {}
|
|
# __members__ returns a copy
|
|
foo = UnscopedEnum.__members__
|
|
foo["bar"] = "baz"
|
|
assert UnscopedEnum.__members__ == {"EOne": UnscopedEnum.EOne, "ETwo": UnscopedEnum.ETwo}
|
|
|
|
# no TypeError exception for unscoped enum ==/!= int comparisons
|
|
y = UnscopedEnum.ETwo
|
|
assert y == 2
|
|
assert y != 3
|
|
|
|
assert int(UnscopedEnum.ETwo) == 2
|
|
assert str(UnscopedEnum(2)) == "UnscopedEnum.ETwo"
|
|
|
|
# order
|
|
assert UnscopedEnum.EOne < UnscopedEnum.ETwo
|
|
assert UnscopedEnum.EOne < 2
|
|
assert UnscopedEnum.ETwo > UnscopedEnum.EOne
|
|
assert UnscopedEnum.ETwo > 1
|
|
assert UnscopedEnum.ETwo <= 2
|
|
assert UnscopedEnum.ETwo >= 2
|
|
assert UnscopedEnum.EOne <= UnscopedEnum.ETwo
|
|
assert UnscopedEnum.EOne <= 2
|
|
assert UnscopedEnum.ETwo >= UnscopedEnum.EOne
|
|
assert UnscopedEnum.ETwo >= 1
|
|
assert not (UnscopedEnum.ETwo < UnscopedEnum.EOne)
|
|
assert not (2 < UnscopedEnum.EOne)
|
|
|
|
|
|
def test_scoped_enum():
|
|
from pybind11_tests import ScopedEnum, test_scoped_enum
|
|
|
|
assert test_scoped_enum(ScopedEnum.Three) == "ScopedEnum::Three"
|
|
z = ScopedEnum.Two
|
|
assert test_scoped_enum(z) == "ScopedEnum::Two"
|
|
|
|
# expected TypeError exceptions for scoped enum ==/!= int comparisons
|
|
with pytest.raises(TypeError):
|
|
assert z == 2
|
|
with pytest.raises(TypeError):
|
|
assert z != 3
|
|
|
|
# order
|
|
assert ScopedEnum.Two < ScopedEnum.Three
|
|
assert ScopedEnum.Three > ScopedEnum.Two
|
|
assert ScopedEnum.Two <= ScopedEnum.Three
|
|
assert ScopedEnum.Two <= ScopedEnum.Two
|
|
assert ScopedEnum.Two >= ScopedEnum.Two
|
|
assert ScopedEnum.Three >= ScopedEnum.Two
|
|
|
|
|
|
def test_implicit_conversion():
|
|
from pybind11_tests import ClassWithUnscopedEnum
|
|
|
|
assert str(ClassWithUnscopedEnum.EMode.EFirstMode) == "EMode.EFirstMode"
|
|
assert str(ClassWithUnscopedEnum.EFirstMode) == "EMode.EFirstMode"
|
|
|
|
f = ClassWithUnscopedEnum.test_function
|
|
first = ClassWithUnscopedEnum.EFirstMode
|
|
second = ClassWithUnscopedEnum.ESecondMode
|
|
|
|
assert f(first) == 1
|
|
|
|
assert f(first) == f(first)
|
|
assert not f(first) != f(first)
|
|
|
|
assert f(first) != f(second)
|
|
assert not f(first) == f(second)
|
|
|
|
assert f(first) == int(f(first))
|
|
assert not f(first) != int(f(first))
|
|
|
|
assert f(first) != int(f(second))
|
|
assert not f(first) == int(f(second))
|
|
|
|
# noinspection PyDictCreation
|
|
x = {f(first): 1, f(second): 2}
|
|
x[f(first)] = 3
|
|
x[f(second)] = 4
|
|
# Hashing test
|
|
assert str(x) == "{EMode.EFirstMode: 3, EMode.ESecondMode: 4}"
|
|
|
|
|
|
def test_binary_operators():
|
|
from pybind11_tests import Flags
|
|
|
|
assert int(Flags.Read) == 4
|
|
assert int(Flags.Write) == 2
|
|
assert int(Flags.Execute) == 1
|
|
assert int(Flags.Read | Flags.Write | Flags.Execute) == 7
|
|
assert int(Flags.Read | Flags.Write) == 6
|
|
assert int(Flags.Read | Flags.Execute) == 5
|
|
assert int(Flags.Write | Flags.Execute) == 3
|
|
assert int(Flags.Write | 1) == 3
|
|
|
|
state = Flags.Read | Flags.Write
|
|
assert (state & Flags.Read) != 0
|
|
assert (state & Flags.Write) != 0
|
|
assert (state & Flags.Execute) == 0
|
|
assert (state & 1) == 0
|
|
|
|
state2 = ~state
|
|
assert state2 == -7
|
|
assert int(state ^ state2) == -1
|
|
|
|
def test_enum_to_int():
|
|
from pybind11_tests import Flags, ClassWithUnscopedEnum
|
|
from pybind11_tests import test_enum_to_int, test_enum_to_uint, test_enum_to_long_long
|
|
|
|
test_enum_to_int(Flags.Read)
|
|
test_enum_to_int(ClassWithUnscopedEnum.EMode.EFirstMode)
|
|
test_enum_to_uint(Flags.Read)
|
|
test_enum_to_uint(ClassWithUnscopedEnum.EMode.EFirstMode)
|
|
test_enum_to_long_long(Flags.Read)
|
|
test_enum_to_long_long(ClassWithUnscopedEnum.EMode.EFirstMode)
|