format test_batchrunner

Rbnstrn · Feb 19, 2021 · 84e85e3 · 84e85e3
1 parent 32d067c
commit 84e85e3
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/test_batchrunner.py b/tests/test_batchrunner.py
@@ -8,7 +8,12 @@
 from mesa import Agent, Model
 from mesa.time import BaseScheduler
 from mesa.datacollection import DataCollector
-from mesa.batchrunner import BatchRunner, FixedBatchRunner, ParameterProduct, ParameterSampler
+from mesa.batchrunner import (
+    BatchRunner,
+    FixedBatchRunner,
+    ParameterProduct,
+    ParameterSampler,
+)
 
 
 NUM_AGENTS = 7
@@ -153,7 +158,9 @@ def model_runs(self):
         if isinstance(self.variable_params, list):
             return len(self.variable_params) * self.iterations
         else:
-            return reduce(mul, map(len, self.variable_params.values())) * self.iterations
+            return (
+                reduce(mul, map(len, self.variable_params.values())) * self.iterations
+            )
 
     def test_model_level_vars(self):
         """