pytype/build_scripts/test_module.py at main · TeleAI-mcp/pytype · GitHub

History

executable file

270 lines (228 loc) · 8.56 KB

Raw

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

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

#! /usr/bin/python -B

"""A convenience script to run tests in a module within the Pytype source tree.

Usage:

$> test_module.py MODULE [-o RESULTS_FILE] [-p] [-s] [-P PATH] [-S]

MODULE is the fully qualified name of a test module within the Pytype source

tree. For example, to run tests in pytype/tests/test_functions.py, specify

the module name as pytype.tests.test_functions.

RESULTS_FILE is the optional path to a file to write the test results into.

By default, only failures are printed to stdout. To see passed tests as

well on stdout, specify the "-p" option.

Be default, failure stack traces are not printed to stdout. To see failure

stack traces on stdout, specify the "-s" option.

By default, the pytype package in the root source directory will be used to run

tests. One can use the -P option to specify the path to a different pytype

package.

Specifying the -S option silences all printing to stdout. It overrides other

print flags (-p and -s).

"""

import argparse

import os

import sys

import traceback

import unittest

import build_utils

def print_messages(options, stdout_msg, output_file_msg):

if not options.silent and stdout_msg:

print(stdout_msg)

if options.output_file and output_file_msg:

options.output_file.write(output_file_msg + "\n")

class StatsCollector:

"""A class which collects stats while running tests."""

def __init__(self, options):

self._options = options

self.class_count = 0

self.method_count = 0

self.error_count = 0

self.fail_count = 0

self.unexpected_success_count = 0

def add_class(self):

self.class_count += 1

def add_method(self, test_result):

self.method_count += 1

self.error_count += len(test_result.errors)

self.fail_count += len(test_result.failures)

self.unexpected_success_count += len(test_result.unexpectedSuccesses)

def report(self):

msg = (

f"\nRan {self.method_count} methods from {self.class_count} classes.\n"

)

msg += "Found %d errors\n" % self.error_count

msg += "Found %d failures\n" % self.fail_count

msg += f"Found {self.unexpected_success_count} unexpected successes\n"

print_messages(self._options, msg, msg)

class ResultReporter:

"""A class which reports results of test runs."""

def __init__(self, options, stats_collector):

self._options = options

self._stats_collector = stats_collector

def _method_info(self, prefix, fq_method_name, group):

common_msg = f"{prefix}: {fq_method_name}"

log_message = f"{common_msg}\n{group[0][1]}"

stdout_message = log_message if self._options.print_st else common_msg

return log_message, stdout_message

def _print_messages(self, stdout_msg, log_msg):

print_messages(self._options, stdout_msg, log_msg)

def report_method(self, fq_method_name, test_result):

self._stats_collector.add_method(test_result)

ret_val = 0

problems_list = [

("ERROR", test_result.errors),

("FAIL", test_result.failures),

("UNEXPECTED PASS", test_result.unexpectedSuccesses),

]

for kind, problems in problems_list:

if problems:

log_msg, stdout_msg = self._method_info(kind, fq_method_name, problems)

ret_val = 1

# There can only be one kind of problem because a different test_result

# object is used for each method.

break

if ret_val == 0:

log_msg = f"PASS: {fq_method_name}"

stdout_msg = log_msg if self._options.print_passes else None

self._print_messages(stdout_msg, log_msg)

return ret_val

def report_class(self, class_name):

self._stats_collector.add_class()

msg = "\nRunning test methods in class %s ..." % (

self._options.fq_mod_name + "." + class_name

)

self._print_messages(msg, msg)

def report_module(self):

msg = f"\nRunning tests in module {self._options.fq_mod_name} ..."

self._print_messages(msg, msg)

def parse_args():

"""Parse the command line options and return the result."""

parser = argparse.ArgumentParser()

parser.add_argument(

"fq_mod_name",

type=str,

metavar="FQ_MOD_NAME",

help="Fully qualified name of the test module to run.",

)

parser.add_argument(

"-P",

"--pytype_path",

type=str,

default=build_utils.PYTYPE_SRC_ROOT,

help="Path in which the pytype package can be found.",

)

parser.add_argument(

"-o", "--output", type=str, help="Path to the results file."

)

parser.add_argument(

"-p",

"--print_passes",

action="store_true",

help="Print information about passing tests to stdout.",

)

parser.add_argument(

"-s",

"--print_st",

action="store_true",

help="Print stack traces of failing tests to stdout.",

)

parser.add_argument(

"-S",

"--silent",

action="store_true",

help="Do not print anything to stdout.",

)

return parser.parse_args()

def run_test_method(method_name, class_object, options, reporter):

"""Run a test method and return 0 on success, 1 on failure."""

test_object = class_object(method_name)

test_object.setUp()

test_result = test_object.defaultTestResult()

test_object.run(test_result)

test_object.tearDown()

fq_method_name = ".".join(

[options.fq_mod_name, class_object.__name__, method_name]

)

return reporter.report_method(fq_method_name, test_result)

def _get_members_list(parent_object):

# We want to create of list of members explicitly as dict.items in

# Python 3 returns an iterator. Hence, we do not want to be in situation where

# in the members dict changes during iteration and raises an exception.

return [(name, getattr(parent_object, name)) for name in dir(parent_object)]

def run_tests_in_class(class_object, options, reporter):

"""Run test methods in a class and return the number of failing methods."""

if getattr(class_object, "__unittest_skip__", False):

return 0

result = 0

class_object.setUpClass()

reporter.report_class(class_object.__name__)

for method_name, method_object in _get_members_list(class_object):

if callable(method_object) and method_name.startswith("test"):

result += run_test_method(method_name, class_object, options, reporter)

class_object.tearDownClass()

return result

def run_tests_in_module(options, reporter):

"""Run test methods in a module and return the number of failing methods.."""

reporter.report_module()

mod_abs_path = os.path.join(

options.pytype_path, options.fq_mod_name.replace(".", os.path.sep) + ".py"

)

if not os.path.exists(mod_abs_path):

msg = f"ERROR: Module not found: {options.fq_mod_name}."

if options.output_file:

options.output_file.write(msg + "\n")

return 1

else:

sys.exit(msg)

try:

if "." in options.fq_mod_name:

fq_pkg_name, _ = options.fq_mod_name.rsplit(".", 1)

mod_object = __import__(options.fq_mod_name, fromlist=[fq_pkg_name])

else:

mod_object = __import__(options.fq_mod_name)

except ImportError:

traceback.print_exc(file=options.output_file)

return 1

result = 0

# Support the load_tests protocol:

# https://docs.python.org/3/library/unittest.html#load-tests-protocol

if hasattr(mod_object, "load_tests"):

top_suite = unittest.suite.TestSuite()

mod_object.load_tests(unittest.TestLoader(), top_suite, None)

suites = [top_suite]

while suites:

suite = suites.pop(0)

for obj in suite:

if isinstance(obj, unittest.TestSuite):

suites.append(obj)

else:

assert isinstance(obj, unittest.TestCase), (obj, type(obj))

result += run_tests_in_class(obj.__class__, options, reporter)

for _, class_object in _get_members_list(mod_object):

if isinstance(class_object, type) and issubclass(

class_object, unittest.TestCase

result += run_tests_in_class(class_object, options, reporter)

return result

def main():

options = parse_args()

# We add path to the pytype package at the beginning of sys.path so that

# it gets picked up before other potential pytype installations present

# already.

sys.path = [options.pytype_path] + sys.path

stats_collector = StatsCollector(options)

reporter = ResultReporter(options, stats_collector)

def run(output_file):

options.output_file = output_file

result = run_tests_in_module(options, reporter)

stats_collector.report()

return result

if options.output:

with open(options.output, "w") as output_file:

result = run(output_file)

else:

result = run(None)

if result != 0:

print(build_utils.failure_msg(options.fq_mod_name, options.output))

sys.exit(1)

else:

print(build_utils.pass_msg(options.fq_mod_name))

if __name__ == "__main__":

main()

pytype/build_scripts/test_module.py at main · TeleAI-mcp/pytype · GitHub

Expand file tree

Breadcrumbs

test_module.py

Latest commit

History

Breadcrumbs

test_module.py

File metadata and controls