Fixed a problem with running in multiprocessor node - incompatible re…

…sults due to using lambdas
mkalinin · Jun 11, 2024 · d88ded5 · d88ded5
1 parent e07e058
commit d88ded5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/tests/generators/fork_choice_generated/instance_generator.py b/tests/generators/fork_choice_generated/instance_generator.py
@@ -53,9 +53,8 @@ def prepare_fn() -> None:
                 def make_cases_fn() -> Iterable[TestCase]:
                     for seed in seeds:
                         mutation_generator = MutatorsGenerator(
-                            spec, seed, number_of_mutations,
-                            lambda: test_fn(fork_name, preset_name, seed, solution),
-                            debug=debug)
+                            fork_name, preset_name, spec, solution,
+                            seed, number_of_mutations, test_fn, debug=debug)
                         for j in range(1 + number_of_mutations):
                             yield TestCase(fork_name=fork_name,
                                         preset_name=preset_name,

diff --git a/tests/generators/fork_choice_generated/mutation_operators.py b/tests/generators/fork_choice_generated/mutation_operators.py
@@ -240,8 +240,11 @@ def mk_mutations(spec, seed, num, test_fn, debug=False):
 
 
 class MutatorsGenerator:
-    def __init__(self, spec, seed, num, test_fn, debug=False):
+    def __init__(self, fork, preset, spec, solution, seed, num, test_fn, debug=False):
+        self.fork = fork
+        self.preset = preset
         self.spec = spec
+        self.solution = solution
         self.seed = seed
         self.num = num
         self.test_fn = test_fn
@@ -250,6 +253,8 @@ def __init__(self, spec, seed, num, test_fn, debug=False):
 
     def next_test_case(self):
         if self.iterator is None:
-            self.iterator = iter(mk_mutations(self.spec, self.seed, self.num, self.test_fn, self.debug))
+            def test_inst_fn():
+                return self.test_fn(self.fork, self.preset, self.seed, self.solution)
+            self.iterator = iter(mk_mutations(self.spec, self.seed, self.num, test_inst_fn, self.debug))
         _, test_case = next(self.iterator)
         return test_case