Fix tests

ISI-MIP · Jul 17, 2024 · 26f3594 · 26f3594
1 parent 488964b
commit 26f3594
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_definitions.py b/tests/test_definitions.py
@@ -58,7 +58,7 @@ def test_variable():
     instance = read_instance(Path('definitions') / 'variable')
 
     for row in instance:
-        sectors = row.get('sectors') + ['other']
+        sectors = row.get('sectors', []) + ['other']
         if sectors:
             for key, value in row.items():
                 if isinstance(value, dict):
@@ -92,8 +92,10 @@ def test_variable_groups():
 
     for variable in variables:
         group = variable.get('group')
-        assert group, variable.get('specifier')
-        assert group in group_specifiers
+        products = variable.get('products')
+        if products is None or 'OutputData' in products:
+            assert group, variable.get('specifier')
+            assert group in group_specifiers
 
 
 def test_nested_simulation_rounds():