Implemented context for pyATF tests to be skipped if not installed

Author: fjwillemsen

test/context.py

@@ -73,6 +73,12 @@
 except ImportError:
     bayes_opt_gpytorch_present = False
 
+try:
+    import pyatf
+    pyatf_present = True
+except ImportError:
+    pyatf_present = False
+
 try:
     from autotuning_methodology.report_experiments import get_strategy_scores
     methodology_present = True
@@ -101,6 +107,7 @@
 skip_if_no_bayesopt_gpytorch = pytest.mark.skipif(not bayes_opt_gpytorch_present, reason="Torch and GPyTorch not installed")
 skip_if_no_bayesopt_botorch = pytest.mark.skipif(not bayes_opt_botorch_present, reason="Torch and BOTorch not installed")
 skip_if_no_hip = pytest.mark.skipif(not hip_present, reason="No HIP Python found")
+skip_if_no_pyatf = pytest.mark.skipif(not pyatf_present, reason="PyATF not installed")
 skip_if_no_methodology = pytest.mark.skipif(not methodology_present, reason="Autotuning Methodology not found")
 
 

test/strategies/test_strategies.py

@@ -8,7 +8,7 @@
 from kernel_tuner.util import InvalidConfig
 from kernel_tuner.interface import strategy_map
 
-from ..context import skip_if_no_bayesopt_botorch, skip_if_no_bayesopt_gpytorch
+from ..context import skip_if_no_bayesopt_botorch, skip_if_no_bayesopt_gpytorch, skip_if_no_pyatf
 
 
 @pytest.fixture
@@ -47,6 +47,8 @@ def vector_add():
         strategies.append(pytest.param(s, marks=skip_if_no_bayesopt_gpytorch))
     elif 'botorch' in s.lower():
         strategies.append(pytest.param(s, marks=skip_if_no_bayesopt_botorch))
+    elif 'pyatf' in s.lower():
+        strategies.append(pytest.param(s, marks=skip_if_no_pyatf))
     else:
         strategies.append(s)
 @pytest.mark.parametrize('strategy', strategies)