tools/testing/selftests/turbostat/defcolumns.py

   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')