1 /*
2 tests/test_modules.cpp -- nested modules, importing modules, and
3 internal references
4
5 Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
6
7 All rights reserved. Use of this source code is governed by a
8 BSD-style license that can be found in the LICENSE file.
9 */
10
11 #include "pybind11_tests.h"
12 #include "constructor_stats.h"
13
TEST_SUBMODULE(modules,m)14 TEST_SUBMODULE(modules, m) {
15 // test_nested_modules
16 // This is intentionally "py::module" to verify it still can be used in place of "py::module_"
17 py::module m_sub = m.def_submodule("subsubmodule");
18 m_sub.def("submodule_func", []() { return "submodule_func()"; });
19
20 // test_reference_internal
21 class A {
22 public:
23 A(int v) : v(v) { print_created(this, v); }
24 ~A() { print_destroyed(this); }
25 A(const A&) { print_copy_created(this); }
26 A& operator=(const A ©) { print_copy_assigned(this); v = copy.v; return *this; }
27 std::string toString() { return "A[" + std::to_string(v) + "]"; }
28 private:
29 int v;
30 };
31 py::class_<A>(m_sub, "A")
32 .def(py::init<int>())
33 .def("__repr__", &A::toString);
34
35 class B {
36 public:
37 B() { print_default_created(this); }
38 ~B() { print_destroyed(this); }
39 B(const B&) { print_copy_created(this); }
40 B& operator=(const B ©) { print_copy_assigned(this); a1 = copy.a1; a2 = copy.a2; return *this; }
41 A &get_a1() { return a1; }
42 A &get_a2() { return a2; }
43
44 A a1{1};
45 A a2{2};
46 };
47 py::class_<B>(m_sub, "B")
48 .def(py::init<>())
49 .def("get_a1", &B::get_a1, "Return the internal A 1", py::return_value_policy::reference_internal)
50 .def("get_a2", &B::get_a2, "Return the internal A 2", py::return_value_policy::reference_internal)
51 .def_readwrite("a1", &B::a1) // def_readonly uses an internal reference return policy by default
52 .def_readwrite("a2", &B::a2);
53
54 // This is intentionally "py::module" to verify it still can be used in place of "py::module_"
55 m.attr("OD") = py::module::import("collections").attr("OrderedDict");
56
57 // test_duplicate_registration
58 // Registering two things with the same name
59 m.def("duplicate_registration", []() {
60 class Dupe1 { };
61 class Dupe2 { };
62 class Dupe3 { };
63 class DupeException { };
64
65 // Go ahead and leak, until we have a non-leaking py::module_ constructor
66 auto dm = py::module_::create_extension_module("dummy", nullptr, new py::module_::module_def);
67 auto failures = py::list();
68
69 py::class_<Dupe1>(dm, "Dupe1");
70 py::class_<Dupe2>(dm, "Dupe2");
71 dm.def("dupe1_factory", []() { return Dupe1(); });
72 py::exception<DupeException>(dm, "DupeException");
73
74 try {
75 py::class_<Dupe1>(dm, "Dupe1");
76 failures.append("Dupe1 class");
77 } catch (std::runtime_error &) {}
78 try {
79 dm.def("Dupe1", []() { return Dupe1(); });
80 failures.append("Dupe1 function");
81 } catch (std::runtime_error &) {}
82 try {
83 py::class_<Dupe3>(dm, "dupe1_factory");
84 failures.append("dupe1_factory");
85 } catch (std::runtime_error &) {}
86 try {
87 py::exception<Dupe3>(dm, "Dupe2");
88 failures.append("Dupe2");
89 } catch (std::runtime_error &) {}
90 try {
91 dm.def("DupeException", []() { return 30; });
92 failures.append("DupeException1");
93 } catch (std::runtime_error &) {}
94 try {
95 py::class_<DupeException>(dm, "DupeException");
96 failures.append("DupeException2");
97 } catch (std::runtime_error &) {}
98
99 return failures;
100 });
101 }
102