From 0bab107e2498c581aee3ebcea6a8fd04878eb19d Mon Sep 17 00:00:00 2001
From: 2Shirt <1923621+2Shirt@users.noreply.github.com>
Date: Wed, 27 Dec 2017 02:11:28 -0700
Subject: [PATCH] Updated results screen formatting

---
 .bin/Scripts/functions/hw_diags.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/.bin/Scripts/functions/hw_diags.py b/.bin/Scripts/functions/hw_diags.py
index 092100ff..a1428dff 100644
--- a/.bin/Scripts/functions/hw_diags.py
+++ b/.bin/Scripts/functions/hw_diags.py
@@ -452,7 +452,9 @@ def show_disk_details(dev):
         for attrib, threshold in sorted(ATTRIBUTES['SMART'].items()):
             if attrib in s_table:
                 print_standard(
-                    '  {:>3}  {:32}'.format(attrib, s_table[attrib]['name']),
+                    '  {:>3}  {:32}'.format(
+                        attrib,
+                        s_table[attrib]['name']).replace('_', ' ').title(),
                     end='', flush=True)
                 raw_str = s_table[attrib]['raw']['string']
                 raw_num = re.sub(r'^(\d+).*$', r'\1', raw_str)
@@ -476,7 +478,7 @@ def show_disk_details(dev):
 
 def show_results():
     clear_screen()
-    print_standard('Results')
+    print_standard('Hardware Diagnostic Results')
     update_progress()
 
     # Set Window layout and show progress
@@ -485,12 +487,12 @@ def show_results():
 
     # Prime95
     if TESTS['Prime95']['Enabled']:
-        print_info('\nPrime95:')
+        print_success('\nPrime95:')
         for log, regex in [
             ['results.txt', r'(error|fail)'],
             ['prime.log', r'completed.*0 errors, 0 warnings']]:
             if log in TESTS['Prime95']:
-                #print_standard(log)
+                print_info('Log: {}'.format(log))
                 lines = [line.strip() for line
                     in TESTS['Prime95'][log].splitlines()
                     if re.search(regex, line, re.IGNORECASE)]
@@ -505,6 +507,7 @@ def show_results():
 
     # NVMe/SMART / badblocks
     if TESTS['NVMe/SMART']['Enabled'] or TESTS['badblocks']['Enabled']:
+        print_success('\nDisks:')
         for name, dev in sorted(TESTS['NVMe/SMART']['Devices'].items()):
             show_disk_details(dev)
             if TESTS['badblocks']['Enabled']: