Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2024-11-29 08:32:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added __contains__ to stl bindings for maps (#1767)

Browse Source 
* Added __contains__ to stl bindings for maps
This commit is contained in:
Blake Thompson 2019-06-10 14:01:11 -05:00 committed by Wenzel Jakob
parent c251434011
commit 30c0352348
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
include/pybind11/stl_bind.h
View File

@ -579,6 +579,15 @@ class_<Map, holder_type> bind_map(handle scope, const std::string &name, Args&&.
return_value_policy::reference_internal // ref + keepalive return_value_policy::reference_internal // ref + keepalive
); );
cl.def("__contains__",
[](Map &m, const KeyType &k) -> bool {
auto it = m.find(k);
if (it == m.end())
return false;
return true;
}
);
// Assignment provided only if the type is copyable // Assignment provided only if the type is copyable
detail::map_assignment<Map, Class_>(cl); detail::map_assignment<Map, Class_>(cl);

2
tests/test_stl.py
View File

@ -56,7 +56,9 @@ def test_map(doc):
"""std::map <-> dict""" """std::map <-> dict"""
d = m.cast_map() d = m.cast_map()
assert d == {"key": "value"} assert d == {"key": "value"}
assert "key" in d
d["key2"] = "value2" d["key2"] = "value2"
assert "key2" in d
assert m.load_map(d) assert m.load_map(d)
assert doc(m.cast_map) == "cast_map() -> Dict[str, str]" assert doc(m.cast_map) == "cast_map() -> Dict[str, str]"