1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
from .support import HPyTest
class TestDict(HPyTest):
def test_Check(self):
mod = self.make_module("""
HPyDef_METH(f, "f", HPyFunc_O)
static HPy f_impl(HPyContext *ctx, HPy self, HPy arg)
{
if (HPyDict_Check(ctx, arg))
return HPy_Dup(ctx, ctx->h_True);
return HPy_Dup(ctx, ctx->h_False);
}
@EXPORT(f)
@INIT
""")
class MyDict(dict):
pass
assert mod.f({}) is True
assert mod.f([]) is False
assert mod.f(MyDict()) is True
def test_New(self):
mod = self.make_module("""
HPyDef_METH(f, "f", HPyFunc_NOARGS)
static HPy f_impl(HPyContext *ctx, HPy self)
{
return HPyDict_New(ctx);
}
@EXPORT(f)
@INIT
""")
assert mod.f() == {}
def test_set_item(self):
mod = self.make_module("""
HPyDef_METH(f, "f", HPyFunc_O)
static HPy f_impl(HPyContext *ctx, HPy self, HPy arg)
{
HPy dict = HPyDict_New(ctx);
if (HPy_IsNull(dict))
return HPy_NULL;
HPy val = HPyLong_FromLong(ctx, 1234);
if (HPy_SetItem(ctx, dict, arg, val) == -1)
return HPy_NULL;
HPy_Close(ctx, val);
return dict;
}
@EXPORT(f)
@INIT
""")
assert mod.f('hello') == {'hello': 1234}
def test_get_item(self):
mod = self.make_module("""
HPyDef_METH(f, "f", HPyFunc_O)
static HPy f_impl(HPyContext *ctx, HPy self, HPy arg)
{
HPy key = HPyUnicode_FromString(ctx, "hello");
if (HPy_IsNull(key))
return HPy_NULL;
HPy val = HPy_GetItem(ctx, arg, key);
HPy_Close(ctx, key);
if (HPy_IsNull(val)) {
HPyErr_Clear(ctx);
return HPy_Dup(ctx, ctx->h_None);
}
return val;
}
@EXPORT(f)
@INIT
""")
assert mod.f({'hello': 1}) == 1
assert mod.f({}) is None
def test_keys(self):
import pytest
mod = self.make_module("""
HPyDef_METH(f, "f", HPyFunc_O)
static HPy f_impl(HPyContext *ctx, HPy self, HPy arg)
{
HPy h_dict = HPy_Is(ctx, arg, ctx->h_None) ? HPy_NULL : arg;
return HPyDict_Keys(ctx, h_dict);
}
@EXPORT(f)
@INIT
""")
class SubDict(dict):
def keys(self):
return [1, 2, 3]
assert mod.f({}) == []
assert mod.f({'hello': 1}) == ['hello']
assert mod.f(SubDict(hello=1)) == ['hello']
with pytest.raises(SystemError):
mod.f(None)
with pytest.raises(SystemError):
mod.f(42)
def test_copy(self):
import pytest
mod = self.make_module("""
HPyDef_METH(f, "f", HPyFunc_O)
static HPy f_impl(HPyContext *ctx, HPy self, HPy arg)
{
HPy h_dict = HPy_Is(ctx, arg, ctx->h_None) ? HPy_NULL : arg;
return HPyDict_Copy(ctx, h_dict);
}
@EXPORT(f)
@INIT
""")
dicts = ({}, {'hello': 1})
for d in dicts:
d_copy = mod.f(d)
assert d_copy == d
assert d_copy is not d
with pytest.raises(SystemError):
mod.f(None)
with pytest.raises(SystemError):
mod.f(42)
|
