test_enum.cpp revision 11986:c12e4625ab56
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 33 }; 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 52 py::enum_<ScopedEnum>(m, "ScopedEnum", py::arithmetic()) 53 .value("Two", ScopedEnum::Two) 54 .value("Three", ScopedEnum::Three); 55 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 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(); 68}); 69