Add tests for the extended enum_ API

Author: aldanor

tests/test_enum.cpp

@@ -47,9 +47,15 @@ test_initializer enums([](py::module &m) {
     py::enum_<UnscopedEnum>(m, "UnscopedEnum", py::arithmetic())
         .value("EOne", EOne)
         .value("ETwo", ETwo)
-        .export_values();
+        .export_values()
+        .into_class()
+        .def("x", [](const UnscopedEnum& e) { return static_cast<int>(e) + 1; })
+        .def_property_readonly("y", [](const UnscopedEnum& e) { return static_cast<int>(e) + 2; })
+        .def_static("a", []() { return 41; })
+        .def_property_readonly_static("b", [](py::object /* unused */) { return 42; });
 
-    py::enum_<ScopedEnum>(m, "ScopedEnum", py::arithmetic())
+    auto scoped_enum = py::enum_<ScopedEnum>(m, "ScopedEnum", py::arithmetic());
+    scoped_enum
         .value("Two", ScopedEnum::Two)
         .value("Three", ScopedEnum::Three);
 

tests/test_enum.py

@@ -40,6 +40,14 @@ def test_unscoped_enum():
     assert not (2 < UnscopedEnum.EOne)
 
 
+def test_extra_defs():
+    from pybind11_tests import UnscopedEnum
+    assert UnscopedEnum.EOne.x() == 2 and UnscopedEnum.ETwo.x() == 3
+    assert UnscopedEnum.EOne.y == 3 and UnscopedEnum.ETwo.y == 4
+    assert UnscopedEnum.a() == 41
+    assert UnscopedEnum.b == 42
+
+
 def test_scoped_enum():
     from pybind11_tests import ScopedEnum, test_scoped_enum