Scheduler: Enable import-chasing in scheduler tests

ecmwf-ifs · Mar 20, 2024 · ad4dde8 · ad4dde8
1 parent adb08bd
commit ad4dde8
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 10 deletions.
diff --git a/tests/test_cmake.py b/tests/test_cmake.py
@@ -63,7 +63,7 @@ def fixture_config(here):
     the file path
     """
     default_config = {
-        'default': {'role': 'kernel', 'expand': True, 'strict': True},
+        'default': {'role': 'kernel', 'expand': True, 'strict': True, 'enable_imports': True},
         'routines': {
             'driverB': {'role': 'driver'},
         }

diff --git a/tests/test_lint/test_linter.py b/tests/test_lint/test_linter.py
@@ -478,7 +478,8 @@ def output(self, handler_reports):
                 'role': 'kernel',
                 'expand': True,
                 'strict': False,
-                'block': ['compute_l2']
+                'block': ['compute_l2'],
+                'enable_imports': True,
             },
             'routines': routines
         }

diff --git a/tests/test_scheduler.py b/tests/test_scheduler.py
@@ -86,7 +86,8 @@ def fixture_config():
             'role': 'kernel',
             'expand': True,
             'strict': True,
-            'disable': ['abort']
+            'disable': ['abort'],
+            'enable_imports': True,
         },
         'routines': {}
     }
@@ -995,15 +996,19 @@ def test_scheduler_dependencies_ignore(here, preprocess, frontend):
     projB = here/'sources/projB'
 
     configA = SchedulerConfig.from_dict({
-        'default': {'role': 'kernel', 'expand': True, 'strict': True},
+        'default': {
+            'role': 'kernel', 'expand': True, 'strict': True, 'enable_imports': True
+        },
         'routines': {
             'driverB': {'role': 'driver'},
             'kernelB': {'ignore': ['ext_driver']},
         }
     })
 
     configB = SchedulerConfig.from_dict({
-        'default': {'role': 'kernel', 'expand': True, 'strict': True},
+        'default': {
+            'role': 'kernel', 'expand': True, 'strict': True, 'enable_imports': True
+        },
         'routines': {
             'ext_driver': {'role': 'kernel'}
         }
@@ -1387,7 +1392,7 @@ def test_scheduler_typebound(here, config, frontend, proj_typebound_dependencies
 
     scheduler = Scheduler(
         paths=proj, seed_routines=['driver'], config=config,
-        full_parse=False, frontend=frontend
+        full_parse=False, frontend=frontend,
     )
 
     assert set(scheduler.items) == set(proj_typebound_dependencies)

diff --git a/transformations/tests/test_data_offload.py b/transformations/tests/test_data_offload.py
@@ -37,6 +37,7 @@ def fixture_config():
             'role': 'kernel',
             'expand': True,
             'strict': True,
+            'enable_imports': True,
         },
     }
 

diff --git a/transformations/tests/test_pool_allocator.py b/transformations/tests/test_pool_allocator.py
@@ -173,6 +173,7 @@ def test_pool_allocator_temporaries(frontend, generate_driver_stack, block_dim,
             'role': 'kernel',
             'expand': True,
             'strict': True,
+            'enable_imports': True,
         },
         'routines': {
             'driver': {'role': 'driver'}
@@ -505,7 +506,8 @@ def test_pool_allocator_temporaries_kernel_sequence(frontend, block_dim, directi
             'mode': 'idem',
             'role': 'kernel',
             'expand': True,
-            'strict': True
+            'strict': True,
+            'enable_imports': True,
         },
         'routines': {
             'driver': {'role': 'driver'}
@@ -789,7 +791,8 @@ def test_pool_allocator_temporaries_kernel_nested(frontend, block_dim, directive
             'mode': 'idem',
             'role': 'kernel',
             'expand': True,
-            'strict': True
+            'strict': True,
+            'enable_imports': True,
         },
         'routines': {
             'driver': {'role': 'driver', 'real_kind': 'jwrb'}
@@ -1020,7 +1023,8 @@ def test_pool_allocator_more_call_checks(frontend, block_dim, caplog):
             'mode': 'idem',
             'role': 'kernel',
             'expand': True,
-            'strict': True
+            'strict': True,
+            'enable_imports': True,
         },
         'routines': {
             'kernel': {}
@@ -1151,7 +1155,8 @@ def test_pool_allocator_args_vs_kwargs(frontend, block_dim):
             'role': 'kernel',
             'expand': True,
             'strict': True,
-            'disable': ['parkind1']
+            'disable': ['parkind1'],
+            'enable_imports': True,
         },
         'routines': {
             'driver': {'role': 'driver'}

diff --git a/transformations/tests/test_transform_derived_types.py b/transformations/tests/test_transform_derived_types.py
@@ -33,6 +33,7 @@ def fixture_config():
             'role': 'kernel',
             'expand': True,
             'strict': True,
+            'enable_imports': True,
         },
         'routines': {
             'driver': {