Minor improvement to inversion stats module

- Check and print error if G-EDF test not enabled
- Clean up output

1 files changed, 6 insertions, 1 deletions

diff --git a/unit-trace b/unit-trace
index 81b6c02..a146874 100755
--- a/unit-trace
+++ b/unit-trace
@@ -110,7 +110,12 @@ if options.stdout is True:
 
 # Print G_EDF inversion statistics
 if options.num_inversions > -1:
-    gedf_inversion_stat_printer.gedf_inversion_stat_printer(stream2,options.num_inversions)
+    if options.gedf is not True:
+        import sys
+        sys.stderr.write("You must enable the G-EDF test module to print" +
+            " G-EDF inversion statistics\n")
+    else:
+        gedf_inversion_stat_printer.gedf_inversion_stat_printer(stream2,options.num_inversions)
 
 # Call visualizer
 if options.visualize is True: