fix cell params test

ntolley · Aug 5, 2024 · d980c6f · d980c6f
1 parent b6534e4
commit d980c6f
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/hnn_core/tests/test_gui.py b/hnn_core/tests/test_gui.py
@@ -283,11 +283,14 @@ def test_gui_init_network():
     assert np.isclose(_single_simulation['net']._inplane_distance, 1.)
     assert np.isclose(_single_simulation['net']._layer_separation, 1307.4)
 
-    default_network_configuration = read_params(hnn_core_root / 'param' / 'default.json')
-    net = jones_2009_model(params=default_network_configuration, add_drives_from_params=True)
+    default_network_configuration = read_params(
+        hnn_core_root / 'param' / 'default.json')
+    net = jones_2009_model(
+        params=default_network_configuration, add_drives_from_params=True)
 
     assert _single_simulation['net'] == net
 
+
 @requires_mpi4py
 @requires_psutil
 def test_gui_run_simulation_mpi(setup_gui):
@@ -778,7 +781,8 @@ def test_gui_cell_params_widgets(setup_gui):
 
     # Check the if the cell params dictionary has been updated
     cell_params = gui.get_cell_parameters_dict()
-    assert (len(cell_params['Geometry']) == 20)
+    assert (len(cell_params['Geometry L2']) == 20)
+    assert (len(cell_params['Geometry L5']) == 22)
     assert (len(cell_params['Synapses']) == 12)
     assert (len(cell_params['Biophysics L2']) == 10)
     assert (len(cell_params['Biophysics L5']) == 20)