Merge pull request #2520 from vkarak/bugfix/detect-reframe-files

[bugfix] Improve detection of ReFrame test files

Author: Vasileios Karakasis

cscs-checks/microbenchmarks/gpu/gpu_burn/gpu_burn_test.py

@@ -3,7 +3,7 @@
 #
 # SPDX-License-Identifier: BSD-3-Clause
 
-import reframe as rfm   # noqa: F501
+import reframe as rfm
 from hpctestlib.microbenchmarks.gpu.gpu_burn import gpu_burn_check
 
 

reframe/frontend/loader.py

@@ -8,6 +8,7 @@
 #
 
 import ast
+import contextlib
 import inspect
 import os
 import sys
@@ -27,7 +28,7 @@ def __init__(self):
 
     @property
     def valid(self):
-        return self._has_import
+        return self._has_import or self._has_regression_test
 
     def visit_Import(self, node):
         for m in node.names:
@@ -38,6 +39,13 @@ def visit_ImportFrom(self, node):
         if node.module is not None and node.module.startswith('reframe'):
             self._has_import = True
 
+    def visit_ClassDef(self, node):
+        for deco in node.decorator_list:
+            with contextlib.suppress(AttributeError):
+                if deco.attr == 'simple_test':
+                    self._has_regression_test = True
+                    break
+
 
 class RegressionCheckLoader:
     def __init__(self, load_path, recurse=False, external_vars=None,