822065cb940ad28ce9f521f828c9e2856a95c629
3 # Copyright 2018 Google, Inc.
5 # Redistribution and use in source and binary forms, with or without
6 # modification, are permitted provided that the following conditions are
7 # met: redistributions of source code must retain the above copyright
8 # notice, this list of conditions and the following disclaimer;
9 # redistributions in binary form must reproduce the above copyright
10 # notice, this list of conditions and the following disclaimer in the
11 # documentation and/or other materials provided with the distribution;
12 # neither the name of the copyright holders nor the names of its
13 # contributors may be used to endorse or promote products derived from
14 # this software without specific prior written permission.
16 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
20 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 from __future__
import print_function
39 import multiprocessing
.pool
45 script_path
= os
.path
.abspath(inspect
.getfile(inspect
.currentframe()))
46 script_dir
= os
.path
.dirname(script_path
)
47 config_path
= os
.path
.join(script_dir
, 'config.py')
49 systemc_rel_path
= 'systemc'
50 tests_rel_path
= os
.path
.join(systemc_rel_path
, 'tests')
51 json_rel_path
= os
.path
.join(tests_rel_path
, 'tests.json')
56 args
= ['scons'] + list(args
)
57 subprocess
.check_call(args
)
62 def __init__(self
, target
, suffix
, build_dir
, props
):
65 self
.build_dir
= build_dir
68 for key
, val
in props
.iteritems():
69 self
.set_prop(key
, val
)
71 def set_prop(self
, key
, val
):
72 setattr(self
, key
, val
)
76 return os
.path
.join(self
.build_dir
, tests_rel_path
, self
.path
)
79 return os
.path
.join(script_dir
, self
.path
)
82 return os
.path
.join(self
.src_dir(), 'golden')
85 return '.'.join([self
.name
, self
.suffix
])
88 return os
.path
.join(self
.dir(), self
.bin())
91 return os
.path
.join(self
.dir(), 'm5out.' + self
.suffix
)
93 def returncode_file(self
):
94 return os
.path
.join(self
.m5out_dir(), 'returncode')
98 test_phase_classes
= {}
100 class TestPhaseMeta(type):
101 def __init__(cls
, name
, bases
, d
):
102 if not d
.pop('abstract', False):
103 test_phase_classes
[d
['name']] = cls
105 super(TestPhaseMeta
, cls
).__init
__(name
, bases
, d
)
107 class TestPhaseBase(object):
108 __metaclass__
= TestPhaseMeta
111 def __init__(self
, main_args
, *args
):
112 self
.main_args
= main_args
115 def __lt__(self
, other
):
116 return self
.number
< other
.number
118 class CompilePhase(TestPhaseBase
):
122 def run(self
, tests
):
123 targets
= list([test
.full_path() for test
in tests
])
124 scons_args
= list(self
.args
) + targets
127 class RunPhase(TestPhaseBase
):
131 def run(self
, tests
):
132 parser
= argparse
.ArgumentParser()
133 parser
.add_argument('--timeout', type=int, metavar
='SECONDS',
134 help='Time limit for each run in seconds.',
136 parser
.add_argument('-j', type=int, default
=1,
137 help='How many tests to run in parallel.')
138 args
= parser
.parse_args(self
.args
)
142 '--kill-after', str(args
.timeout
* 2),
148 cmd
.extend(timeout_cmd
)
151 '-red', test
.m5out_dir(),
152 '--listener-mode=off',
156 # Ensure the output directory exists.
157 if not os
.path
.exists(test
.m5out_dir()):
158 os
.makedirs(test
.m5out_dir())
160 subprocess
.check_call(cmd
)
161 except subprocess
.CalledProcessError
, error
:
162 returncode
= error
.returncode
165 with
open(test
.returncode_file(), 'w') as rc
:
166 rc
.write('%d\n' % returncode
)
168 runnable
= filter(lambda t
: not t
.compile_only
, tests
)
170 map(run_test
, runnable
)
172 tp
= multiprocessing
.pool
.ThreadPool(args
.j
)
173 map(lambda t
: tp
.apply_async(run_test
, (t
,)), runnable
)
177 class Checker(object):
178 def __init__(self
, ref
, test
, tag
):
184 with
open(self
.text
) as test_f
, open(self
.ref
) as ref_f
:
185 return test_f
.read() == ref_f
.read()
187 class LogChecker(Checker
):
188 def merge_filts(*filts
):
189 filts
= map(lambda f
: '(' + f
+ ')', filts
)
190 filts
= '|'.join(filts
)
191 return re
.compile(filts
, flags
=re
.MULTILINE
)
193 ref_filt
= merge_filts(
194 r
'^\nInfo: /OSCI/SystemC: Simulation stopped by user.\n',
195 r
'^SystemC Simulation\n',
196 r
'^\nWarning: .*\nIn file: .*\n'
198 test_filt
= merge_filts(
199 r
'^Global frequency set at \d* ticks per second\n'
202 def __init__(self
, ref
, test
, tag
, out_dir
):
203 super(LogChecker
, self
).__init
__(ref
, test
, tag
)
204 self
.out_dir
= out_dir
206 def apply_filters(self
, data
, filts
):
207 re
.sub(filt
, '', data
)
210 test_file
= os
.path
.basename(self
.test
)
211 ref_file
= os
.path
.basename(self
.ref
)
212 with
open(self
.test
) as test_f
, open(self
.ref
) as ref_f
:
213 test
= re
.sub(self
.test_filt
, '', test_f
.read())
214 ref
= re
.sub(self
.ref_filt
, '', ref_f
.read())
216 diff_file
= '.'.join([ref_file
, 'diff'])
217 diff_path
= os
.path
.join(self
.out_dir
, diff_file
)
218 with
open(diff_path
, 'w') as diff_f
:
219 for line
in difflib
.unified_diff(
220 ref
.splitlines(True), test
.splitlines(True),
227 class VerifyPhase(TestPhaseBase
):
231 def reset_status(self
):
235 def passed(self
, test
):
236 self
._passed
.append(test
)
238 def failed(self
, test
, cause
, note
=''):
239 test
.set_prop('note', note
)
240 self
._failed
.setdefault(cause
, []).append(test
)
242 def print_status(self
):
243 total_passed
= len(self
._passed
)
244 total_failed
= sum(map(len, self
._failed
.values()))
246 print('Passed: {passed:4} - Failed: {failed:4}'.format(
247 passed
=total_passed
, failed
=total_failed
))
249 def write_result_file(self
, path
):
251 'passed': map(lambda t
: t
.props
, self
._passed
),
253 cause
: map(lambda t
: t
.props
, tests
) for
254 cause
, tests
in self
._failed
.iteritems()
257 with
open(path
, 'w') as rf
:
258 json
.dump(results
, rf
)
260 def print_results(self
):
263 for path
in sorted(list([ t
.path
for t
in self
._passed
])):
270 for cause
, tests
in sorted(self
._failed
.items()):
271 block
= ' ' + cause
.capitalize() + ':\n'
272 for test
in sorted(tests
, key
=lambda t
: t
.path
):
273 block
+= ' ' + test
.path
275 block
+= ' - ' + test
.note
279 print('\n'.join(causes
))
281 def run(self
, tests
):
282 parser
= argparse
.ArgumentParser()
283 result_opts
= parser
.add_mutually_exclusive_group()
284 result_opts
.add_argument('--result-file', action
='store_true',
285 help='Create a results.json file in the current directory.')
286 result_opts
.add_argument('--result-file-at', metavar
='PATH',
287 help='Create a results json file at the given path.')
288 parser
.add_argument('--print-results', action
='store_true',
289 help='Print a list of tests that passed or failed')
290 args
= parser
.parse_args(self
.args
)
294 runnable
= filter(lambda t
: not t
.compile_only
, tests
)
295 compile_only
= filter(lambda t
: t
.compile_only
, tests
)
297 for test
in compile_only
:
298 if os
.path
.exists(test
.full_path()):
301 self
.failed(test
, 'compile failed')
303 for test
in runnable
:
304 with
open(test
.returncode_file()) as rc
:
305 returncode
= int(rc
.read())
307 if returncode
== 124:
308 self
.failed(test
, 'time out')
310 elif returncode
!= 0:
311 self
.failed(test
, 'abort')
314 out_dir
= test
.m5out_dir()
316 Diff
= collections
.namedtuple(
317 'Diff', 'ref, test, tag, ref_filter')
321 log_file
= '.'.join([test
.name
, 'log'])
322 log_path
= os
.path
.join(test
.golden_dir(), log_file
)
323 simout_path
= os
.path
.join(out_dir
, 'simout')
324 if not os
.path
.exists(simout_path
):
325 self
.failed(test
, 'no log output')
326 if os
.path
.exists(log_path
):
327 diffs
.append(LogChecker(
328 log_path
, simout_path
, log_file
, out_dir
))
330 failed_diffs
= filter(lambda d
: not d
.check(), diffs
)
332 tags
= map(lambda d
: d
.tag
, failed_diffs
)
333 self
.failed(test
, 'failed diffs', ' '.join(tags
))
338 if args
.print_results
:
345 result_path
= os
.path
.join(os
.getcwd(), 'results.json')
346 elif args
.result_file_at
:
347 result_path
= args
.result_file_at
350 self
.write_result_file(result_path
)
353 parser
= argparse
.ArgumentParser(description
='SystemC test utility')
355 parser
.add_argument('build_dir', metavar
='BUILD_DIR',
356 help='The build directory (ie. build/ARM).')
358 parser
.add_argument('--update-json', action
='store_true',
359 help='Update the json manifest of tests.')
361 parser
.add_argument('--flavor', choices
=['debug', 'opt', 'fast'],
363 help='Flavor of binary to test.')
365 parser
.add_argument('--list', action
='store_true',
366 help='List the available tests')
368 filter_opts
= parser
.add_mutually_exclusive_group()
369 filter_opts
.add_argument('--filter', default
='True',
370 help='Python expression which filters tests based '
371 'on their properties')
372 filter_opts
.add_argument('--filter-file', default
=None,
373 type=argparse
.FileType('r'),
374 help='Same as --filter, but read from a file')
376 def collect_phases(args
):
377 phase_groups
= [list(g
) for k
, g
in
378 itertools
.groupby(args
, lambda x
: x
!= '--phase') if k
]
379 main_args
= parser
.parse_args(phase_groups
[0][1:])
382 for group
in phase_groups
[1:]:
385 raise RuntimeException('Phase %s specified more than once' % name
)
386 phase
= test_phase_classes
[name
]
387 phases
.append(phase(main_args
, *group
[1:]))
389 return main_args
, phases
391 main_args
, phases
= collect_phases(sys
.argv
)
395 CompilePhase(main_args
),
397 VerifyPhase(main_args
)
402 json_path
= os
.path
.join(main_args
.build_dir
, json_rel_path
)
404 if main_args
.update_json
:
405 scons(os
.path
.join(json_path
))
407 with
open(json_path
) as f
:
408 test_data
= json
.load(f
)
410 if main_args
.filter_file
:
411 f
= main_args
.filter_file
412 filt
= compile(f
.read(), f
.name
, 'eval')
414 filt
= compile(main_args
.filter, '<string>', 'eval')
417 target
: props
for (target
, props
) in
418 test_data
.iteritems() if eval(filt
, dict(props
))
422 for target
, props
in sorted(filtered_tests
.iteritems()):
423 print('%s.%s' % (target
, main_args
.flavor
))
424 for key
, val
in props
.iteritems():
425 print(' %s: %s' % (key
, val
))
426 print('Total tests: %d' % len(filtered_tests
))
428 tests_to_run
= list([
429 Test(target
, main_args
.flavor
, main_args
.build_dir
, props
) for
430 target
, props
in sorted(filtered_tests
.iteritems())
434 phase
.run(tests_to_run
)