GeezerVison is showing github.com — pythoncapi-compat/tests/test_pythoncapi_compat.py at main · python/pythoncapi-compat

Expand file tree

test_pythoncapi_compat.py

Copy path

More file actions

Latest commit

History

238 lines (187 loc) · 5.77 KB

test_pythoncapi_compat.py

Copy path

File metadata and controls

238 lines (187 loc) · 5.77 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

#!/usr/bin/python3

"""

Run the test suite.

Usage::

python3 run_tests.py

python3 run_tests.py -v # verbose mode

"""

import argparse

import faulthandler

import gc

import os.path

import shutil

import subprocess

import sys

import sysconfig

# test.utils

import setup

from utils import run_command, get_output

# Windows uses MSVC compiler

MSVC = (os.name == "nt")

# C++ is only supported on Python 3.6 and newer

TEST_CXX = (sys.version_info >= (3, 6))

if not MSVC:

C_TESTS = [

("test_pythoncapi_compat_cext_c99", "C99"),

("test_pythoncapi_compat_cext_c11", "C11"),

]

CXX_TESTS = [

("test_pythoncapi_compat_cpp03ext", "C++03"),

("test_pythoncapi_compat_cpp11ext", "C++11"),

("test_pythoncapi_compat_cpp14ext", "C++14"),

("test_pythoncapi_compat_cpp17ext", "C++17"),

("test_pythoncapi_compat_cpp20ext", "C++20"),

]

else:

C_TESTS = [

("test_pythoncapi_compat_cext_c11", "C11"),

]

CXX_TESTS = [

("test_pythoncapi_compat_cppext", "C++"),

("test_pythoncapi_compat_cpp14ext", "C++14"),

]

VERBOSE = False

def display_title(title):

if not VERBOSE:

return

title = "%s: %s" % (python_version(), title)

print(title)

print("=" * len(title))

print()

sys.stdout.flush()

def build_ext(build_dir):

display_title("Build test extensions")

cmd = [sys.executable, "-u", "setup.py", "build", "--build-base", build_dir]

if VERBOSE:

run_command(cmd)

print()

else:

exitcode, stdout = get_output(cmd)

if exitcode:

print(stdout.rstrip())

sys.exit(exitcode)

def import_tests(build_dir, module_name):

pythonpath = None

for name in os.listdir(build_dir):

if name.startswith('lib.'):

pythonpath = os.path.join(build_dir, name)

if not pythonpath:

raise Exception("Failed to find the build directory")

old_sys_path = list(sys.path)

try:

sys.path.append(pythonpath)

return __import__(module_name)

finally:

sys.path[:] = old_sys_path

def _run_tests(tests, verbose):

for name, test_func in tests:

if verbose:

print("%s()" % name)

sys.stdout.flush()

test_func()

_HAS_CLEAR_INTERNAL_CACHES = hasattr(sys, '_clear_internal_caches')

_HAS_CLEAR_TYPE_CACHE = hasattr(sys, '_clear_type_cache')

def _refleak_cleanup():

if _HAS_CLEAR_INTERNAL_CACHES:

sys._clear_internal_caches()

elif _HAS_CLEAR_TYPE_CACHE:

sys._clear_type_cache()

gc.collect()

def _check_refleak(test_func, verbose):

nrun = 6

for i in range(1, nrun + 1):

if verbose:

if i > 1:

print()

print("Run %s/%s:" % (i, nrun))

sys.stdout.flush()

init_refcnt = sys.gettotalrefcount()

test_func()

_refleak_cleanup()

diff = sys.gettotalrefcount() - init_refcnt

if i > 3 and diff:

raise AssertionError("refcnt leak, diff: %s" % diff)

def python_version():

ver = sys.version_info

build = 'debug' if hasattr(sys, 'gettotalrefcount') else 'release'

if hasattr(sys, 'implementation'):

python_impl = sys.implementation.name

if python_impl == 'cpython':

python_impl = 'CPython'

elif python_impl == 'pypy':

python_impl = 'PyPy'

else:

if "PyPy" in sys.version:

python_impl = "PyPy"

else:

python_impl = 'Python'

pyver = "%s.%s" % (ver.major, ver.minor)

if ver >= (3, 13):

if sysconfig.get_config_var('Py_GIL_DISABLED'):

# Free-threaded build

pyver += "t"

return "%s %s (%s build)" % (python_impl, pyver, build)

def run_tests(build_dir, module_name, std):

lang = std.upper() if std else None

if VERBOSE:

print("")

title = f"Test {module_name}"

if lang:

titlte = f"{title} ({lang})"

display_title(title)

testmod = import_tests(build_dir, module_name)

if VERBOSE:

empty_line = False

for attr in ('__cplusplus', 'PY_VERSION', 'PY_VERSION_HEX',

'PYPY_VERSION', 'PYPY_VERSION_NUM', 'Py_LIMITED_API'):

try:

value = getattr(testmod, attr)

except AttributeError:

pass

else:

if attr in ('PY_VERSION_HEX', 'Py_LIMITED_API', 'PYPY_VERSION_NUM'):

value = "0x%x" % value

print("%s: %s" % (attr, value))

empty_line = True

if empty_line:

print()

check_refleak = hasattr(sys, 'gettotalrefcount')

tests = [(name, getattr(testmod, name))

for name in dir(testmod)

if name.startswith("test")]

def test_func():

_run_tests(tests, VERBOSE)

if check_refleak:

_check_refleak(test_func, VERBOSE)

else:

test_func()

if VERBOSE:

print()

msg = f"{python_version()}, {module_name}: {len(tests)} tests succeeded!"

if check_refleak:

msg += " (no reference leak detected)"

print(msg)

# Unload the extension module

testmod = None

del sys.modules[module_name]

def parse_args():

parser = argparse.ArgumentParser()

parser.add_argument('-v', '--verbose',

action='store_true') # on/off flag

parser.add_argument('build_dir')

return parser.parse_args()

def main():

faulthandler.enable()

global VERBOSE

args = parse_args()

VERBOSE = args.verbose

build_dir = args.build_dir

src_dir = os.path.dirname(__file__)

if src_dir:

os.chdir(src_dir)

build_ext(build_dir)

tests = setup.C_TESTS + setup.CXX_TESTS

for module_name, std, limited in tests:

run_tests(build_dir, module_name, std)

if __name__ == "__main__":

main()