From 049cd849b8f8168808c808eaa0c837b3904e408d Mon Sep 17 00:00:00 2001 From: Vaclav Petras Date: Wed, 16 Sep 2020 10:37:29 -0400 Subject: [PATCH] Be more explicit about value, fix Black --- tests/test_simulation_function.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_simulation_function.py b/tests/test_simulation_function.py index 3a631dd..cbe824a 100644 --- a/tests/test_simulation_function.py +++ b/tests/test_simulation_function.py @@ -18,7 +18,8 @@ def test_simulation_function(): weather_coefficient = np.array([[0.6, 0.8, 0.7], [0.2, 0.8, 0.5]], dtype=float_type) latency_period_steps = 0 - exposed = [np.array([[0, 0, 0], [0, 0, 0]], dtype=int_type)] * (latency_period_steps + 1) + exposed_size = latency_period_steps + 1 + exposed = [np.array([[0, 0, 0], [0, 0, 0]], dtype=int_type)] * exposed_size a = pypops.test_simulation( random_seed=42,