From 488e78d6deca020f940985d0a21358349ebb9321 Mon Sep 17 00:00:00 2001 From: Lena Salfenmoser Date: Tue, 5 Sep 2023 14:42:34 +0200 Subject: [PATCH] Update test_autochunk.py --- tests/test_autochunk.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test_autochunk.py b/tests/test_autochunk.py index d9d2e0b9..29f82ab9 100644 --- a/tests/test_autochunk.py +++ b/tests/test_autochunk.py @@ -86,15 +86,17 @@ def test_onstep_input_autochunk(self): ous = np.zeros((model.params["N"], duration_dt)) # prepare input - inp_x = np.zeros((model.params["N"], duration_dt)) - inp_y = np.zeros((model.params["N"], duration_dt)) + inp_exc_current = np.zeros((model.params["N"], duration_dt)) + inp_inh_current = np.zeros((model.params["N"], duration_dt)) + inp_exc_rate = np.zeros((model.params["N"], duration_dt)) + inp_inh_rate = np.zeros((model.params["N"], duration_dt)) for n in range(model.params["N"]): fr = 1 - inp_x[n, :] = np.sin(np.linspace(0, fr * 2 * np.pi, duration_dt)) * 0.1 + inp_exc_current[n, :] = np.sin(np.linspace(0, fr * 2 * np.pi, duration_dt)) * 0.1 for i in range(duration_dt): - inputs = [inp_x[:, i], inp_y[:, i]] + inputs = [inp_exc_current[:, i], inp_inh_current[:, i], inp_exc_rate[:, i], inp_inh_rate[:, i]] model.autochunk(inputs=inputs, append_outputs=True)