|
Packit |
534379 |
/*
|
|
Packit |
534379 |
tests/test_pytypes.cpp -- Python type casters
|
|
Packit |
534379 |
|
|
Packit |
534379 |
Copyright (c) 2017 Wenzel Jakob <wenzel.jakob@epfl.ch>
|
|
Packit |
534379 |
|
|
Packit |
534379 |
All rights reserved. Use of this source code is governed by a
|
|
Packit |
534379 |
BSD-style license that can be found in the LICENSE file.
|
|
Packit |
534379 |
*/
|
|
Packit |
534379 |
|
|
Packit |
534379 |
#include "pybind11_tests.h"
|
|
Packit |
534379 |
|
|
Packit |
534379 |
|
|
Packit |
534379 |
TEST_SUBMODULE(pytypes, m) {
|
|
Packit |
534379 |
// test_list
|
|
Packit |
534379 |
m.def("get_list", []() {
|
|
Packit |
534379 |
py::list list;
|
|
Packit |
534379 |
list.append("value");
|
|
Packit |
534379 |
py::print("Entry at position 0:", list[0]);
|
|
Packit |
534379 |
list[0] = py::str("overwritten");
|
|
Packit |
534379 |
list.insert(0, "inserted-0");
|
|
Packit |
534379 |
list.insert(2, "inserted-2");
|
|
Packit |
534379 |
return list;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("print_list", [](py::list list) {
|
|
Packit |
534379 |
int index = 0;
|
|
Packit |
534379 |
for (auto item : list)
|
|
Packit |
534379 |
py::print("list item {}: {}"_s.format(index++, item));
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_set
|
|
Packit |
534379 |
m.def("get_set", []() {
|
|
Packit |
534379 |
py::set set;
|
|
Packit |
534379 |
set.add(py::str("key1"));
|
|
Packit |
534379 |
set.add("key2");
|
|
Packit |
534379 |
set.add(std::string("key3"));
|
|
Packit |
534379 |
return set;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("print_set", [](py::set set) {
|
|
Packit |
534379 |
for (auto item : set)
|
|
Packit |
534379 |
py::print("key:", item);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("set_contains", [](py::set set, py::object key) {
|
|
Packit |
534379 |
return set.contains(key);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("set_contains", [](py::set set, const char* key) {
|
|
Packit |
534379 |
return set.contains(key);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_dict
|
|
Packit |
534379 |
m.def("get_dict", []() { return py::dict("key"_a="value"); });
|
|
Packit |
534379 |
m.def("print_dict", [](py::dict dict) {
|
|
Packit |
534379 |
for (auto item : dict)
|
|
Packit |
534379 |
py::print("key: {}, value={}"_s.format(item.first, item.second));
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("dict_keyword_constructor", []() {
|
|
Packit |
534379 |
auto d1 = py::dict("x"_a=1, "y"_a=2);
|
|
Packit |
534379 |
auto d2 = py::dict("z"_a=3, **d1);
|
|
Packit |
534379 |
return d2;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("dict_contains", [](py::dict dict, py::object val) {
|
|
Packit |
534379 |
return dict.contains(val);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
m.def("dict_contains", [](py::dict dict, const char* val) {
|
|
Packit |
534379 |
return dict.contains(val);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_str
|
|
Packit |
534379 |
m.def("str_from_string", []() { return py::str(std::string("baz")); });
|
|
Packit |
534379 |
m.def("str_from_bytes", []() { return py::str(py::bytes("boo", 3)); });
|
|
Packit |
534379 |
m.def("str_from_object", [](const py::object& obj) { return py::str(obj); });
|
|
Packit |
534379 |
m.def("repr_from_object", [](const py::object& obj) { return py::repr(obj); });
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("str_format", []() {
|
|
Packit |
534379 |
auto s1 = "{} + {} = {}"_s.format(1, 2, 3);
|
|
Packit |
534379 |
auto s2 = "{a} + {b} = {c}"_s.format("a"_a=1, "b"_a=2, "c"_a=3);
|
|
Packit |
534379 |
return py::make_tuple(s1, s2);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_bytes
|
|
Packit |
534379 |
m.def("bytes_from_string", []() { return py::bytes(std::string("foo")); });
|
|
Packit |
534379 |
m.def("bytes_from_str", []() { return py::bytes(py::str("bar", 3)); });
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_capsule
|
|
Packit |
534379 |
m.def("return_capsule_with_destructor", []() {
|
|
Packit |
534379 |
py::print("creating capsule");
|
|
Packit |
534379 |
return py::capsule([]() {
|
|
Packit |
534379 |
py::print("destructing capsule");
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("return_capsule_with_destructor_2", []() {
|
|
Packit |
534379 |
py::print("creating capsule");
|
|
Packit |
534379 |
return py::capsule((void *) 1234, [](void *ptr) {
|
|
Packit |
534379 |
py::print("destructing capsule: {}"_s.format((size_t) ptr));
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("return_capsule_with_name_and_destructor", []() {
|
|
Packit |
534379 |
auto capsule = py::capsule((void *) 1234, "pointer type description", [](PyObject *ptr) {
|
|
Packit |
534379 |
if (ptr) {
|
|
Packit |
534379 |
auto name = PyCapsule_GetName(ptr);
|
|
Packit |
534379 |
py::print("destructing capsule ({}, '{}')"_s.format(
|
|
Packit |
534379 |
(size_t) PyCapsule_GetPointer(ptr, name), name
|
|
Packit |
534379 |
));
|
|
Packit |
534379 |
}
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
void *contents = capsule;
|
|
Packit |
534379 |
py::print("created capsule ({}, '{}')"_s.format((size_t) contents, capsule.name()));
|
|
Packit |
534379 |
return capsule;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_accessors
|
|
Packit |
534379 |
m.def("accessor_api", [](py::object o) {
|
|
Packit |
534379 |
auto d = py::dict();
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["basic_attr"] = o.attr("basic_attr");
|
|
Packit |
534379 |
|
|
Packit |
534379 |
auto l = py::list();
|
|
Packit |
534379 |
for (const auto &item : o.attr("begin_end")) {
|
|
Packit |
534379 |
l.append(item);
|
|
Packit |
534379 |
}
|
|
Packit |
534379 |
d["begin_end"] = l;
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["operator[object]"] = o.attr("d")["operator[object]"_s];
|
|
Packit |
534379 |
d["operator[char *]"] = o.attr("d")["operator[char *]"];
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["attr(object)"] = o.attr("sub").attr("attr_obj");
|
|
Packit |
534379 |
d["attr(char *)"] = o.attr("sub").attr("attr_char");
|
|
Packit |
534379 |
try {
|
|
Packit |
534379 |
o.attr("sub").attr("missing").ptr();
|
|
Packit |
534379 |
} catch (const py::error_already_set &) {
|
|
Packit |
534379 |
d["missing_attr_ptr"] = "raised"_s;
|
|
Packit |
534379 |
}
|
|
Packit |
534379 |
try {
|
|
Packit |
534379 |
o.attr("missing").attr("doesn't matter");
|
|
Packit |
534379 |
} catch (const py::error_already_set &) {
|
|
Packit |
534379 |
d["missing_attr_chain"] = "raised"_s;
|
|
Packit |
534379 |
}
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["is_none"] = o.attr("basic_attr").is_none();
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["operator()"] = o.attr("func")(1);
|
|
Packit |
534379 |
d["operator*"] = o.attr("func")(*o.attr("begin_end"));
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// Test implicit conversion
|
|
Packit |
534379 |
py::list implicit_list = o.attr("begin_end");
|
|
Packit |
534379 |
d["implicit_list"] = implicit_list;
|
|
Packit |
534379 |
py::dict implicit_dict = o.attr("__dict__");
|
|
Packit |
534379 |
d["implicit_dict"] = implicit_dict;
|
|
Packit |
534379 |
|
|
Packit |
534379 |
return d;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("tuple_accessor", [](py::tuple existing_t) {
|
|
Packit |
534379 |
try {
|
|
Packit |
534379 |
existing_t[0] = 1;
|
|
Packit |
534379 |
} catch (const py::error_already_set &) {
|
|
Packit |
534379 |
// --> Python system error
|
|
Packit |
534379 |
// Only new tuples (refcount == 1) are mutable
|
|
Packit |
534379 |
auto new_t = py::tuple(3);
|
|
Packit |
534379 |
for (size_t i = 0; i < new_t.size(); ++i) {
|
|
Packit |
534379 |
new_t[i] = i;
|
|
Packit |
534379 |
}
|
|
Packit |
534379 |
return new_t;
|
|
Packit |
534379 |
}
|
|
Packit |
534379 |
return py::tuple();
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("accessor_assignment", []() {
|
|
Packit |
534379 |
auto l = py::list(1);
|
|
Packit |
534379 |
l[0] = 0;
|
|
Packit |
534379 |
|
|
Packit |
534379 |
auto d = py::dict();
|
|
Packit |
534379 |
d["get"] = l[0];
|
|
Packit |
534379 |
auto var = l[0];
|
|
Packit |
534379 |
d["deferred_get"] = var;
|
|
Packit |
534379 |
l[0] = 1;
|
|
Packit |
534379 |
d["set"] = l[0];
|
|
Packit |
534379 |
var = 99; // this assignment should not overwrite l[0]
|
|
Packit |
534379 |
d["deferred_set"] = l[0];
|
|
Packit |
534379 |
d["var"] = var;
|
|
Packit |
534379 |
|
|
Packit |
534379 |
return d;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_constructors
|
|
Packit |
534379 |
m.def("default_constructors", []() {
|
|
Packit |
534379 |
return py::dict(
|
|
Packit |
534379 |
"str"_a=py::str(),
|
|
Packit |
534379 |
"bool"_a=py::bool_(),
|
|
Packit |
534379 |
"int"_a=py::int_(),
|
|
Packit |
534379 |
"float"_a=py::float_(),
|
|
Packit |
534379 |
"tuple"_a=py::tuple(),
|
|
Packit |
534379 |
"list"_a=py::list(),
|
|
Packit |
534379 |
"dict"_a=py::dict(),
|
|
Packit |
534379 |
"set"_a=py::set()
|
|
Packit |
534379 |
);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("converting_constructors", [](py::dict d) {
|
|
Packit |
534379 |
return py::dict(
|
|
Packit |
534379 |
"str"_a=py::str(d["str"]),
|
|
Packit |
534379 |
"bool"_a=py::bool_(d["bool"]),
|
|
Packit |
534379 |
"int"_a=py::int_(d["int"]),
|
|
Packit |
534379 |
"float"_a=py::float_(d["float"]),
|
|
Packit |
534379 |
"tuple"_a=py::tuple(d["tuple"]),
|
|
Packit |
534379 |
"list"_a=py::list(d["list"]),
|
|
Packit |
534379 |
"dict"_a=py::dict(d["dict"]),
|
|
Packit |
534379 |
"set"_a=py::set(d["set"]),
|
|
Packit |
534379 |
"memoryview"_a=py::memoryview(d["memoryview"])
|
|
Packit |
534379 |
);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("cast_functions", [](py::dict d) {
|
|
Packit |
534379 |
// When converting between Python types, obj.cast<T>() should be the same as T(obj)
|
|
Packit |
534379 |
return py::dict(
|
|
Packit |
534379 |
"str"_a=d["str"].cast<py::str>(),
|
|
Packit |
534379 |
"bool"_a=d["bool"].cast<py::bool_>(),
|
|
Packit |
534379 |
"int"_a=d["int"].cast<py::int_>(),
|
|
Packit |
534379 |
"float"_a=d["float"].cast<py::float_>(),
|
|
Packit |
534379 |
"tuple"_a=d["tuple"].cast<py::tuple>(),
|
|
Packit |
534379 |
"list"_a=d["list"].cast<py::list>(),
|
|
Packit |
534379 |
"dict"_a=d["dict"].cast<py::dict>(),
|
|
Packit |
534379 |
"set"_a=d["set"].cast<py::set>(),
|
|
Packit |
534379 |
"memoryview"_a=d["memoryview"].cast<py::memoryview>()
|
|
Packit |
534379 |
);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("get_implicit_casting", []() {
|
|
Packit |
534379 |
py::dict d;
|
|
Packit |
534379 |
d["char*_i1"] = "abc";
|
|
Packit |
534379 |
const char *c2 = "abc";
|
|
Packit |
534379 |
d["char*_i2"] = c2;
|
|
Packit |
534379 |
d["char*_e"] = py::cast(c2);
|
|
Packit |
534379 |
d["char*_p"] = py::str(c2);
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["int_i1"] = 42;
|
|
Packit |
534379 |
int i = 42;
|
|
Packit |
534379 |
d["int_i2"] = i;
|
|
Packit |
534379 |
i++;
|
|
Packit |
534379 |
d["int_e"] = py::cast(i);
|
|
Packit |
534379 |
i++;
|
|
Packit |
534379 |
d["int_p"] = py::int_(i);
|
|
Packit |
534379 |
|
|
Packit |
534379 |
d["str_i1"] = std::string("str");
|
|
Packit |
534379 |
std::string s2("str1");
|
|
Packit |
534379 |
d["str_i2"] = s2;
|
|
Packit |
534379 |
s2[3] = '2';
|
|
Packit |
534379 |
d["str_e"] = py::cast(s2);
|
|
Packit |
534379 |
s2[3] = '3';
|
|
Packit |
534379 |
d["str_p"] = py::str(s2);
|
|
Packit |
534379 |
|
|
Packit |
534379 |
py::list l(2);
|
|
Packit |
534379 |
l[0] = 3;
|
|
Packit |
534379 |
l[1] = py::cast(6);
|
|
Packit |
534379 |
l.append(9);
|
|
Packit |
534379 |
l.append(py::cast(12));
|
|
Packit |
534379 |
l.append(py::int_(15));
|
|
Packit |
534379 |
|
|
Packit |
534379 |
return py::dict(
|
|
Packit |
534379 |
"d"_a=d,
|
|
Packit |
534379 |
"l"_a=l
|
|
Packit |
534379 |
);
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
// test_print
|
|
Packit |
534379 |
m.def("print_function", []() {
|
|
Packit |
534379 |
py::print("Hello, World!");
|
|
Packit |
534379 |
py::print(1, 2.0, "three", true, std::string("-- multiple args"));
|
|
Packit |
534379 |
auto args = py::make_tuple("and", "a", "custom", "separator");
|
|
Packit |
534379 |
py::print("*args", *args, "sep"_a="-");
|
|
Packit |
534379 |
py::print("no new line here", "end"_a=" -- ");
|
|
Packit |
534379 |
py::print("next print");
|
|
Packit |
534379 |
|
|
Packit |
534379 |
auto py_stderr = py::module::import("sys").attr("stderr");
|
|
Packit |
534379 |
py::print("this goes to stderr", "file"_a=py_stderr);
|
|
Packit |
534379 |
|
|
Packit |
534379 |
py::print("flush", "flush"_a=true);
|
|
Packit |
534379 |
|
|
Packit |
534379 |
py::print("{a} + {b} = {c}"_s.format("a"_a="py::print", "b"_a="str.format", "c"_a="this"));
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("print_failure", []() { py::print(42, UnregisteredType()); });
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("hash_function", [](py::object obj) { return py::hash(obj); });
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("test_number_protocol", [](py::object a, py::object b) {
|
|
Packit |
534379 |
py::list l;
|
|
Packit |
534379 |
l.append(a.equal(b));
|
|
Packit |
534379 |
l.append(a.not_equal(b));
|
|
Packit |
534379 |
l.append(a < b);
|
|
Packit |
534379 |
l.append(a <= b);
|
|
Packit |
534379 |
l.append(a > b);
|
|
Packit |
534379 |
l.append(a >= b);
|
|
Packit |
534379 |
l.append(a + b);
|
|
Packit |
534379 |
l.append(a - b);
|
|
Packit |
534379 |
l.append(a * b);
|
|
Packit |
534379 |
l.append(a / b);
|
|
Packit |
534379 |
l.append(a | b);
|
|
Packit |
534379 |
l.append(a & b);
|
|
Packit |
534379 |
l.append(a ^ b);
|
|
Packit |
534379 |
l.append(a >> b);
|
|
Packit |
534379 |
l.append(a << b);
|
|
Packit |
534379 |
return l;
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
|
|
Packit |
534379 |
m.def("test_list_slicing", [](py::list a) {
|
|
Packit |
534379 |
return a[py::slice(0, -1, 2)];
|
|
Packit |
534379 |
});
|
|
Packit |
534379 |
}
|