Add label for test mode in hardware diagnostics

scripts/wk/hw/diags.py

@@ -79,7 +79,7 @@ PLATFORM = std.PLATFORM
 # Classes
 class State():
   """Object for tracking hardware diagnostic data."""
-  def __init__(self):
+  def __init__(self, test_mode=False):
     self.disks = []
     self.layout = cfg.hw.TMUX_LAYOUT.copy()
     self.log_dir = None
@@ -87,6 +87,8 @@ class State():
     self.system = None
     self.test_groups = []
     self.top_text = std.color_string('Hardware Diagnostics', 'GREEN')
+    if test_mode:
+      self.top_text += std.color_string(' (Test Mode)', 'YELLOW')
 
     # Init tmux and start a background process to maintain layout
     self.init_tmux()
@@ -760,7 +762,7 @@ def main() -> None:
   # Init
   atexit.register(tmux.kill_all_panes)
   menu = build_menu(cli_mode=args['--cli'], quick_mode=args['--quick'])
-  state = State()
+  state = State(test_mode=args['--test-mode'])
 
   # Quick Mode
   if args['--quick']: