add test

Author: benvanwerkhoven

kernel_tuner/util.py

@@ -1117,8 +1117,9 @@ def compile_restrictions(
     return noncompiled_restrictions + compiled_restrictions
 
 def check_matching_problem_size(cached_problem_size, problem_size):
+    """ check if requested problem size matches the problem size in the cache"""
     if not (np.array(cached_problem_size) == np.array(problem_size)).all():
-        raise ValueError(f"Cannot load cache which contains results for different problem_size, cache: {cached_data['problem_size']}, requested: {kernel_options.problem_size}")
+        raise ValueError(f"Cannot load cache which contains results for different problem_size, cache: {cached_problem_size}, requested: {problem_size}")
 
 def process_cache(cache, kernel_options, tuning_options, runner):
     """Cache file for storing tuned configurations.

test/test_util_functions.py

@@ -726,6 +726,19 @@ def test_parse_restrictions():
     assert all(param in tune_params for param in params)
 
 
+def test_check_matching_problem_size():
+    # these should error
+    with pytest.raises(ValueError):
+        check_matching_problem_size(42, 1000)
+    with pytest.raises(ValueError):
+        check_matching_problem_size([42,1], 42)
+    # these should not error
+    check_matching_problem_size(1000, (1000,))
+    check_matching_problem_size([1000], 1000)
+    check_matching_problem_size(1000, 1000)
+    check_matching_problem_size(1000, [1000])
+
+
 def test_convert_constraint_lambdas():
 
     restrictions = [lambda p: 32 <= p["block_size_x"]*p["block_size_y"] <= 1024,