c79706ff4c
Change-Id: I52e4fc9ebf2f59da57d8cf8f3e37cc79598c2f5f Signed-off-by: Andreas Sandberg <andreas.sandberg@arm.com> Reviewed-by: Andreas Hansson <andreas.hansson@arm.com> Reviewed-by: Curtis Dunham <curtis.dunham@arm.com> Reviewed-on: https://gem5-review.googlesource.com/2229 Reviewed-by: Tony Gutierrez <anthony.gutierrez@amd.com> Reviewed-by: Jason Lowe-Power <jason@lowepower.com> Reviewed-by: Pierre-Yves Péneau <pierre-yves.peneau@lirmm.fr>
109 lines
3.2 KiB
Python
109 lines
3.2 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"
|
|
|
|
# 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
|