1 #!/bin/env python3 2 # SPDX-License-Identifier: GPL-2.0 3 4 import subprocess 5 from shutil import which 6 7 turbostat = which('turbostat') 8 if turbostat is None: 9 print('Could not find turbostat binary') 10 exit(1) 11 12 timeout = which('timeout') 13 if timeout is None: 14 print('Could not find timeout binary') 15 exit(1) 16 17 proc_turbostat = subprocess.run([turbostat, '--list'], capture_output = True) 18 if proc_turbostat.returncode != 0: 19 print(f'turbostat failed with {proc_turbostat.returncode}') 20 exit(1) 21 22 # 23 # By default --list reports also "usec" and "Time_Of_Day_Seconds" columns 24 # which are only visible when running with --debug. 25 # 26 expected_columns_debug = proc_turbostat.stdout.replace(b',', b'\t').strip() 27 expected_columns = expected_columns_debug.replace(b'usec\t', b'').replace(b'Time_Of_Day_Seconds\t', b'').replace(b'X2APIC\t', b'').replace(b'APIC\t', b'') 28 29 # 30 # Run turbostat with no options for 10 seconds and send SIGINT 31 # 32 timeout_argv = [timeout, '--preserve-status', '-s', 'SIGINT', '-k', '3', '1s'] 33 turbostat_argv = [turbostat, '-i', '0.250'] 34 35 print(f'Running turbostat with {turbostat_argv=}... ', end = '', flush = True) 36 proc_turbostat = subprocess.run(timeout_argv + turbostat_argv, capture_output = True) 37 if proc_turbostat.returncode != 0: 38 print(f'turbostat failed with {proc_turbostat.returncode}') 39 exit(1) 40 actual_columns = proc_turbostat.stdout.split(b'\n')[0] 41 if expected_columns != actual_columns: 42 print(f'turbostat column check failed\n{expected_columns=}\n{actual_columns=}') 43 exit(1) 44 print('OK') 45 46 # 47 # Same, but with --debug 48 # 49 turbostat_argv.append('--debug') 50 51 print(f'Running turbostat with {turbostat_argv=}... ', end = '', flush = True) 52 proc_turbostat = subprocess.run(timeout_argv + turbostat_argv, capture_output = True) 53 if proc_turbostat.returncode != 0: 54 print(f'turbostat failed with {proc_turbostat.returncode}') 55 exit(1) 56 actual_columns = proc_turbostat.stdout.split(b'\n')[0] 57 if expected_columns_debug != actual_columns: 58 print(f'turbostat column check failed\n{expected_columns_debug=}\n{actual_columns=}') 59 exit(1) 60 print('OK')
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.