61 lines · python
1#!/bin/env python32# SPDX-License-Identifier: GPL-2.03 4import subprocess5from shutil import which6 7turbostat = which('turbostat')8if turbostat is None:9 print('Could not find turbostat binary')10 exit(1)11 12timeout = which('timeout')13if timeout is None:14 print('Could not find timeout binary')15 exit(1)16 17proc_turbostat = subprocess.run([turbostat, '--list'], capture_output = True)18if 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" columns24# which are only visible when running with --debug.25#26expected_columns_debug = proc_turbostat.stdout.replace(b',', b'\t').strip()27expected_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 SIGINT31#32timeout_argv = [timeout, '--preserve-status', '-s', 'SIGINT', '-k', '3', '1s']33turbostat_argv = [turbostat, '-i', '0.250']34 35print(f'Running turbostat with {turbostat_argv=}... ', end = '', flush = True)36proc_turbostat = subprocess.run(timeout_argv + turbostat_argv, capture_output = True)37if proc_turbostat.returncode != 0:38 print(f'turbostat failed with {proc_turbostat.returncode}')39 exit(1)40actual_columns = proc_turbostat.stdout.split(b'\n')[0]41if expected_columns != actual_columns:42 print(f'turbostat column check failed\n{expected_columns=}\n{actual_columns=}')43 exit(1)44print('OK')45 46#47# Same, but with --debug48#49turbostat_argv.append('--debug')50 51print(f'Running turbostat with {turbostat_argv=}... ', end = '', flush = True)52proc_turbostat = subprocess.run(timeout_argv + turbostat_argv, capture_output = True)53if proc_turbostat.returncode != 0:54 print(f'turbostat failed with {proc_turbostat.returncode}')55 exit(1)56actual_columns = proc_turbostat.stdout.split(b'\n')[0]57if expected_columns_debug != actual_columns:58 print(f'turbostat column check failed\n{expected_columns_debug=}\n{actual_columns=}')59 exit(1)60print('OK')61