# -*- coding: utf-8 -*- import pytest from pybind11_tests import enums as m def test_unscoped_enum(): assert str(m.UnscopedEnum.EOne) == "UnscopedEnum.EOne" assert str(m.UnscopedEnum.ETwo) == "UnscopedEnum.ETwo" assert str(m.EOne) == "UnscopedEnum.EOne" assert repr(m.UnscopedEnum.EOne) == "" assert repr(m.UnscopedEnum.ETwo) == "" assert repr(m.EOne) == "" # name property assert m.UnscopedEnum.EOne.name == "EOne" assert m.UnscopedEnum.ETwo.name == "ETwo" assert m.EOne.name == "EOne" # name readonly with pytest.raises(AttributeError): m.UnscopedEnum.EOne.name = "" # name returns a copy foo = m.UnscopedEnum.EOne.name foo = "bar" assert m.UnscopedEnum.EOne.name == "EOne" # __members__ property assert m.UnscopedEnum.__members__ == { "EOne": m.UnscopedEnum.EOne, "ETwo": m.UnscopedEnum.ETwo, "EThree": m.UnscopedEnum.EThree, } # __members__ readonly with pytest.raises(AttributeError): m.UnscopedEnum.__members__ = {} # __members__ returns a copy foo = m.UnscopedEnum.__members__ foo["bar"] = "baz" assert m.UnscopedEnum.__members__ == { "EOne": m.UnscopedEnum.EOne, "ETwo": m.UnscopedEnum.ETwo, "EThree": m.UnscopedEnum.EThree, } for docstring_line in """An unscoped enumeration Members: EOne : Docstring for EOne ETwo : Docstring for ETwo EThree : Docstring for EThree""".split( "\n" ): assert docstring_line in m.UnscopedEnum.__doc__ # Unscoped enums will accept ==/!= int comparisons y = m.UnscopedEnum.ETwo assert y == 2 assert 2 == y assert y != 3 assert 3 != y # Compare with None assert y != None # noqa: E711 assert not (y == None) # noqa: E711 # Compare with an object assert y != object() assert not (y == object()) # Compare with string assert y != "2" assert "2" != y assert not ("2" == y) assert not (y == "2") with pytest.raises(TypeError): y < object() with pytest.raises(TypeError): y <= object() with pytest.raises(TypeError): y > object() with pytest.raises(TypeError): y >= object() with pytest.raises(TypeError): y | object() with pytest.raises(TypeError): y & object() with pytest.raises(TypeError): y ^ object() assert int(m.UnscopedEnum.ETwo) == 2 assert str(m.UnscopedEnum(2)) == "UnscopedEnum.ETwo" # order assert m.UnscopedEnum.EOne < m.UnscopedEnum.ETwo assert m.UnscopedEnum.EOne < 2 assert m.UnscopedEnum.ETwo > m.UnscopedEnum.EOne assert m.UnscopedEnum.ETwo > 1 assert m.UnscopedEnum.ETwo <= 2 assert m.UnscopedEnum.ETwo >= 2 assert m.UnscopedEnum.EOne <= m.UnscopedEnum.ETwo assert m.UnscopedEnum.EOne <= 2 assert m.UnscopedEnum.ETwo >= m.UnscopedEnum.EOne assert m.UnscopedEnum.ETwo >= 1 assert not (m.UnscopedEnum.ETwo < m.UnscopedEnum.EOne) assert not (2 < m.UnscopedEnum.EOne) # arithmetic assert m.UnscopedEnum.EOne & m.UnscopedEnum.EThree == m.UnscopedEnum.EOne assert m.UnscopedEnum.EOne | m.UnscopedEnum.ETwo == m.UnscopedEnum.EThree assert m.UnscopedEnum.EOne ^ m.UnscopedEnum.EThree == m.UnscopedEnum.ETwo def test_scoped_enum(): assert m.test_scoped_enum(m.ScopedEnum.Three) == "ScopedEnum::Three" z = m.ScopedEnum.Two assert m.test_scoped_enum(z) == "ScopedEnum::Two" # Scoped enums will

from pybind11_tests import union_ as m


def test_union():
    instance = m.TestUnion()

    instance.as_uint = 10
    assert instance.as_int == 10