Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 1 | /* |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 2 | tests/test_inheritance.cpp -- inheritance, automatic upcasting for polymorphic types |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 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 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 10 | #include "pybind11_tests.h" |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 11 | |
| 12 | class Pet { |
| 13 | public: |
| 14 | Pet(const std::string &name, const std::string &species) |
| 15 | : m_name(name), m_species(species) {} |
| 16 | std::string name() const { return m_name; } |
| 17 | std::string species() const { return m_species; } |
| 18 | private: |
| 19 | std::string m_name; |
| 20 | std::string m_species; |
| 21 | }; |
| 22 | |
| 23 | class Dog : public Pet { |
| 24 | public: |
| 25 | Dog(const std::string &name) : Pet(name, "dog") {} |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 26 | std::string bark() const { return "Woof!"; } |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 27 | }; |
| 28 | |
| 29 | class Rabbit : public Pet { |
| 30 | public: |
| 31 | Rabbit(const std::string &name) : Pet(name, "parrot") {} |
| 32 | }; |
| 33 | |
Jason Rhinelander | 6b52c83 | 2016-09-06 12:27:00 -0400 | [diff] [blame] | 34 | class Hamster : public Pet { |
| 35 | public: |
| 36 | Hamster(const std::string &name) : Pet(name, "rodent") {} |
| 37 | }; |
| 38 | |
Dean Moldovan | 08cbe8d | 2017-02-15 21:10:25 +0100 | [diff] [blame] | 39 | class Chimera : public Pet { |
| 40 | Chimera() : Pet("Kimmy", "chimera") {} |
| 41 | }; |
| 42 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 43 | std::string pet_name_species(const Pet &pet) { |
| 44 | return pet.name() + " is a " + pet.species(); |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 45 | } |
| 46 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 47 | std::string dog_bark(const Dog &dog) { |
| 48 | return dog.bark(); |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 49 | } |
| 50 | |
| 51 | |
| 52 | struct BaseClass { virtual ~BaseClass() {} }; |
| 53 | struct DerivedClass1 : BaseClass { }; |
| 54 | struct DerivedClass2 : BaseClass { }; |
| 55 | |
Pim Schellart | cc88aae | 2017-01-31 10:52:11 -0500 | [diff] [blame] | 56 | struct MismatchBase1 { }; |
| 57 | struct MismatchDerived1 : MismatchBase1 { }; |
| 58 | |
| 59 | struct MismatchBase2 { }; |
| 60 | struct MismatchDerived2 : MismatchBase2 { }; |
| 61 | |
Jason Rhinelander | 52f4be8 | 2016-09-03 14:54:22 -0400 | [diff] [blame] | 62 | test_initializer inheritance([](py::module &m) { |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 63 | py::class_<Pet> pet_class(m, "Pet"); |
| 64 | pet_class |
| 65 | .def(py::init<std::string, std::string>()) |
| 66 | .def("name", &Pet::name) |
| 67 | .def("species", &Pet::species); |
| 68 | |
| 69 | /* One way of declaring a subclass relationship: reference parent's class_ object */ |
| 70 | py::class_<Dog>(m, "Dog", pet_class) |
| 71 | .def(py::init<std::string>()); |
| 72 | |
| 73 | /* Another way of declaring a subclass relationship: reference parent's C++ type */ |
Wenzel Jakob | 8e5dceb | 2016-09-11 20:00:40 +0900 | [diff] [blame] | 74 | py::class_<Rabbit, Pet>(m, "Rabbit") |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 75 | .def(py::init<std::string>()); |
| 76 | |
Jason Rhinelander | 6b52c83 | 2016-09-06 12:27:00 -0400 | [diff] [blame] | 77 | /* And another: list parent in class template arguments */ |
| 78 | py::class_<Hamster, Pet>(m, "Hamster") |
| 79 | .def(py::init<std::string>()); |
| 80 | |
Dean Moldovan | 08cbe8d | 2017-02-15 21:10:25 +0100 | [diff] [blame] | 81 | py::class_<Chimera, Pet>(m, "Chimera"); |
| 82 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 83 | m.def("pet_name_species", pet_name_species); |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 84 | m.def("dog_bark", dog_bark); |
| 85 | |
| 86 | py::class_<BaseClass>(m, "BaseClass").def(py::init<>()); |
| 87 | py::class_<DerivedClass1>(m, "DerivedClass1").def(py::init<>()); |
| 88 | py::class_<DerivedClass2>(m, "DerivedClass2").def(py::init<>()); |
| 89 | |
| 90 | m.def("return_class_1", []() -> BaseClass* { return new DerivedClass1(); }); |
| 91 | m.def("return_class_2", []() -> BaseClass* { return new DerivedClass2(); }); |
Jason Rhinelander | 0e48977 | 2016-09-11 18:41:28 -0400 | [diff] [blame] | 92 | m.def("return_class_n", [](int n) -> BaseClass* { |
| 93 | if (n == 1) return new DerivedClass1(); |
| 94 | if (n == 2) return new DerivedClass2(); |
| 95 | return new BaseClass(); |
| 96 | }); |
Jason Rhinelander | b3f3d79 | 2016-07-18 16:43:18 -0400 | [diff] [blame] | 97 | m.def("return_none", []() -> BaseClass* { return nullptr; }); |
Dean Moldovan | b4498ef | 2016-10-23 14:50:08 +0200 | [diff] [blame] | 98 | |
| 99 | m.def("test_isinstance", [](py::list l) { |
| 100 | struct Unregistered { }; // checks missing type_info code path |
| 101 | |
| 102 | return py::make_tuple( |
| 103 | py::isinstance<py::tuple>(l[0]), |
| 104 | py::isinstance<py::dict>(l[1]), |
| 105 | py::isinstance<Pet>(l[2]), |
| 106 | py::isinstance<Pet>(l[3]), |
| 107 | py::isinstance<Dog>(l[4]), |
| 108 | py::isinstance<Rabbit>(l[5]), |
| 109 | py::isinstance<Unregistered>(l[6]) |
| 110 | ); |
| 111 | }); |
Pim Schellart | cc88aae | 2017-01-31 10:52:11 -0500 | [diff] [blame] | 112 | |
| 113 | m.def("test_mismatched_holder_type_1", []() { |
| 114 | auto m = py::module::import("__main__"); |
| 115 | py::class_<MismatchBase1, std::shared_ptr<MismatchBase1>>(m, "MismatchBase1"); |
| 116 | py::class_<MismatchDerived1, MismatchBase1>(m, "MismatchDerived1"); |
| 117 | }); |
| 118 | m.def("test_mismatched_holder_type_2", []() { |
| 119 | auto m = py::module::import("__main__"); |
| 120 | py::class_<MismatchBase2>(m, "MismatchBase2"); |
| 121 | py::class_<MismatchDerived2, std::shared_ptr<MismatchDerived2>, MismatchBase2>(m, "MismatchDerived2"); |
| 122 | }); |
Jason Rhinelander | 52f4be8 | 2016-09-03 14:54:22 -0400 | [diff] [blame] | 123 | }); |