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 |
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
|
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 }; |
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 }; |
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 }; |
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} |
|