1/* 2 tests/test_enums.cpp -- enumerations 3 4 Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch> 5 6 All rights reserved. Use of this source code is governed by a 7 BSD-style license that can be found in the LICENSE file. 8*/ 9 10#include "pybind11_tests.h" 11
| 1/* 2 tests/test_enums.cpp -- enumerations 3 4 Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch> 5 6 All rights reserved. Use of this source code is governed by a 7 BSD-style license that can be found in the LICENSE file. 8*/ 9 10#include "pybind11_tests.h" 11
|
12enum UnscopedEnum { 13 EOne = 1, 14 ETwo 15}; 16 17enum class ScopedEnum { 18 Two = 2, 19 Three 20}; 21 22enum Flags { 23 Read = 4, 24 Write = 2, 25 Execute = 1 26}; 27 28class ClassWithUnscopedEnum { 29public: 30 enum EMode { 31 EFirstMode = 1, 32 ESecondMode
| 12TEST_SUBMODULE(enums, m) { 13 // test_unscoped_enum 14 enum UnscopedEnum { 15 EOne = 1, 16 ETwo
|
33 };
| 17 };
|
34 35 static EMode test_function(EMode mode) { 36 return mode; 37 } 38}; 39 40std::string test_scoped_enum(ScopedEnum z) { 41 return "ScopedEnum::" + std::string(z == ScopedEnum::Two ? "Two" : "Three"); 42} 43 44test_initializer enums([](py::module &m) { 45 m.def("test_scoped_enum", &test_scoped_enum); 46
| |
47 py::enum_<UnscopedEnum>(m, "UnscopedEnum", py::arithmetic()) 48 .value("EOne", EOne) 49 .value("ETwo", ETwo) 50 .export_values(); 51
| 18 py::enum_<UnscopedEnum>(m, "UnscopedEnum", py::arithmetic()) 19 .value("EOne", EOne) 20 .value("ETwo", ETwo) 21 .export_values(); 22
|
| 23 // test_scoped_enum 24 enum class ScopedEnum { 25 Two = 2, 26 Three 27 };
|
52 py::enum_<ScopedEnum>(m, "ScopedEnum", py::arithmetic()) 53 .value("Two", ScopedEnum::Two) 54 .value("Three", ScopedEnum::Three); 55
| 28 py::enum_<ScopedEnum>(m, "ScopedEnum", py::arithmetic()) 29 .value("Two", ScopedEnum::Two) 30 .value("Three", ScopedEnum::Three); 31
|
| 32 m.def("test_scoped_enum", [](ScopedEnum z) { 33 return "ScopedEnum::" + std::string(z == ScopedEnum::Two ? "Two" : "Three"); 34 }); 35 36 // test_binary_operators 37 enum Flags { 38 Read = 4, 39 Write = 2, 40 Execute = 1 41 };
|
56 py::enum_<Flags>(m, "Flags", py::arithmetic()) 57 .value("Read", Flags::Read) 58 .value("Write", Flags::Write) 59 .value("Execute", Flags::Execute) 60 .export_values(); 61
| 42 py::enum_<Flags>(m, "Flags", py::arithmetic()) 43 .value("Read", Flags::Read) 44 .value("Write", Flags::Write) 45 .value("Execute", Flags::Execute) 46 .export_values(); 47
|
| 48 // test_implicit_conversion 49 class ClassWithUnscopedEnum { 50 public: 51 enum EMode { 52 EFirstMode = 1, 53 ESecondMode 54 }; 55 56 static EMode test_function(EMode mode) { 57 return mode; 58 } 59 };
|
62 py::class_<ClassWithUnscopedEnum> exenum_class(m, "ClassWithUnscopedEnum"); 63 exenum_class.def_static("test_function", &ClassWithUnscopedEnum::test_function); 64 py::enum_<ClassWithUnscopedEnum::EMode>(exenum_class, "EMode") 65 .value("EFirstMode", ClassWithUnscopedEnum::EFirstMode) 66 .value("ESecondMode", ClassWithUnscopedEnum::ESecondMode) 67 .export_values();
| 60 py::class_<ClassWithUnscopedEnum> exenum_class(m, "ClassWithUnscopedEnum"); 61 exenum_class.def_static("test_function", &ClassWithUnscopedEnum::test_function); 62 py::enum_<ClassWithUnscopedEnum::EMode>(exenum_class, "EMode") 63 .value("EFirstMode", ClassWithUnscopedEnum::EFirstMode) 64 .value("ESecondMode", ClassWithUnscopedEnum::ESecondMode) 65 .export_values();
|
68});
| 66 67 // test_enum_to_int 68 m.def("test_enum_to_int", [](int) { }); 69 m.def("test_enum_to_uint", [](uint32_t) { }); 70 m.def("test_enum_to_long_long", [](long long) { }); 71}
|
| |