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
|
# mode: run
from cpython.contextvars cimport (
PyContextVar_New, PyContextVar_New_with_default,
get_value, get_value_no_default,
)
NOTHING = object()
CVAR = PyContextVar_New("cvar", NULL)
CVAR_WITH_DEFAULT = PyContextVar_New_with_default("cvar_wd", "DEFAULT")
import contextvars
PYCVAR = contextvars.ContextVar("pycvar")
def disable_for_pypy737(f):
import sys
# will be fixed in PyPy 7.3.8
if hasattr(sys, 'pypy_version_info') and sys.pypy_version_info < (7,3,8):
return None
return f
@disable_for_pypy737
def test_get_value(var, default=NOTHING):
"""
>>> test_get_value(CVAR)
>>> test_get_value(CVAR, "default")
'default'
>>> test_get_value(PYCVAR)
>>> test_get_value(PYCVAR, "default")
'default'
>>> test_get_value(CVAR_WITH_DEFAULT)
'DEFAULT'
>>> test_get_value(CVAR_WITH_DEFAULT, "default")
'DEFAULT'
"""
return get_value(var, default) if default is not NOTHING else get_value(var)
@disable_for_pypy737
def test_get_value_no_default(var, default=NOTHING):
"""
>>> test_get_value_no_default(CVAR)
>>> test_get_value_no_default(CVAR, "default")
'default'
>>> test_get_value_no_default(PYCVAR)
>>> test_get_value_no_default(PYCVAR, "default")
'default'
>>> test_get_value_no_default(CVAR_WITH_DEFAULT)
>>> test_get_value_no_default(CVAR_WITH_DEFAULT, "default")
'default'
"""
return get_value_no_default(var, default) if default is not NOTHING else get_value_no_default(var)
__test__ = {}
|
