Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-02 19:39:49 +09:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3271 from Jack-R-lantern/dict_key/__contains__

Browse Source 
Add contains in dict_keys
This commit is contained in:
Jeong YunWon
2021-10-12 02:37:13 +09:00
committed by GitHub
parent ec91e6e730 ab43f91fcb
commit fd71647a37
2 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
extra_tests/snippets/builtin_dict.py
View File

@@ -36,4 +36,15 @@ assert ('a') not in d.items()
assert ('a', 123) in d.items()
assert ('b', 456) in d.items()
assert ('a', 123, 3) not in d.items()
assert ('a', 123, 'b', 456) not in d.items()
assert ('a', 123, 'b', 456) not in d.items()
d = {1: 10, "a": "ABC", (3,4): 5}
assert 1 in d.keys()
assert (1) in d.keys()
assert "a" in d.keys()
assert (3,4) in d.keys()
assert () not in d.keys()
assert 10 not in d.keys()
assert (1, 10) not in d.keys()
assert "abc" not in d.keys()
assert ((3,4),5) not in d.keys()

7
vm/src/builtins/dict.rs
View File

@@ -956,7 +956,12 @@ trait ViewSetOps: DictView {
impl ViewSetOps for PyDictKeys {}
#[pyimpl(with(DictView, Comparable, Iterable, ViewSetOps))]
impl PyDictKeys {}
impl PyDictKeys {
#[pymethod(magic)]
fn contains(zelf: PyRef<Self>, key: PyObjectRef, vm: &VirtualMachine) -> PyResult<bool> {
zelf.dict().contains(key, vm)
}
}
impl ViewSetOps for PyDictItems {}
#[pyimpl(with(DictView, Comparable, Iterable, ViewSetOps))]