Follow up for "[lit] Include unexecuted tests in xUnit report"

The extended test fails on bots that still run lit tests with Python 2.

https://reviews.llvm.org/D81316

diff --git a/llvm/utils/lit/tests/Inputs/xunit-output/dummy_format.py b/llvm/utils/lit/tests/Inputs/xunit-output/dummy_format.py
index 94410d2..59e75eb 100644 (file)
--- a/llvm/utils/lit/tests/Inputs/xunit-output/dummy_format.py
+++ b/llvm/utils/lit/tests/Inputs/xunit-output/dummy_format.py
@@ -23,9 +23,10 @@ class DummyFormat(lit.formats.FileBasedTest):
         result = lit.Test.Result(getattr(lit.Test, result_code),
                                  result_output)
 
-        required_feature = cfg.get('global', 'required_feature', fallback=None)
-        if required_feature:
-            test.requires.append(required_feature)
+        if cfg.has_option('global', 'required_feature'):
+            required_feature = cfg.get('global', 'required_feature')
+            if required_feature:
+                test.requires.append(required_feature)
 
         # Load additional metrics.
         for key,value_str in cfg.items('results'):