[MRG] BUG: Update gid_ranges after using clear_drives

doc/whats_new.rst

@@ -94,6 +94,9 @@ Bug
 - Fix drive seeding so that event times are unique across multiple trials,
   by `Nick Tolley`_ in :gh:`810`.
 
+- Fix bug in :func:`~hnn_core/network/clear_drives` where gid_ranges are not
+  accurately updated, by `Nick Tolley`_ in :gh:`812`.
+
 API
 ~~~
 - :func:`~hnn_core.CellResponse.write` and :func:`~hnn_core.Cell_response.read_spikes`

hnn_core/network.py

@@ -1374,12 +1374,17 @@ def clear_connectivity(self):
 
     def clear_drives(self):
         """Remove all drives defined in Network.connectivity"""
-        connectivity = list()
-        for conn in self.connectivity:
-            if conn['src_type'] not in self.external_drives.keys():
-                connectivity.append(conn)
+        self.connectivity = [conn for conn in self.connectivity if
+                             conn['src_type'] if conn['src_type'] not
+                             in self.external_drives.keys()]
+
+        for cell_type in list(self.gid_ranges.keys()):
+            if cell_type in self.external_drives:
+                self._n_gids -= len(self.gid_ranges[cell_type])
+                del self.gid_ranges[cell_type]
+                del self.pos_dict[cell_type]
+
         self.external_drives = dict()
-        self.connectivity = connectivity
 
     def add_electrode_array(self, name, electrode_pos, *, conductivity=0.3,
                             method='psa', min_distance=0.5):

hnn_core/tests/test_drives.py

@@ -98,8 +98,8 @@ def test_external_drive_times():
 def test_drive_seeds(setup_net):
     """Test that unique spike times are generated across trials"""
     net = setup_net
-    weights_ampa = {'L2_basket': 0.000003, 'L2_pyramidal': 1.438840,
-                    'L5_basket': 0.008958, 'L5_pyramidal': 0.684013}
+    weights_ampa = {'L2_basket': 0.3, 'L2_pyramidal': 0.3,
+                    'L5_basket': 0.3, 'L5_pyramidal': 0.3}
     synaptic_delays = {'L2_basket': 0.1, 'L2_pyramidal': 0.1,
                        'L5_basket': 1., 'L5_pyramidal': 1.}
     net.add_evoked_drive(
@@ -116,6 +116,30 @@ def test_drive_seeds(setup_net):
     assert ~np.any(np.allclose(trial1_spikes, trial2_spikes))
 
 
+def test_clear_drives(setup_net):
+    """Test clearing drives updates Network"""
+    net = setup_net
+    weights_ampa = {'L5_pyramidal': 0.3}
+    synaptic_delays = {'L5_pyramidal': 1.}
+
+    n_gids = net._n_gids
+    net.add_evoked_drive(
+        'prox', mu=40, sigma=8.33, numspikes=1,
+        weights_ampa=weights_ampa, location='proximal',
+        synaptic_delays=synaptic_delays, cell_specific=True)
+
+    assert len(net.external_drives) > 0
+    assert 'prox' in net.external_drives
+    assert 'prox' in net.gid_ranges
+    assert net._n_gids == n_gids + len(net.gid_ranges['L5_pyramidal'])
+
+    net.clear_drives()
+    assert len(net.external_drives) == 0
+    assert 'prox' not in net.external_drives
+    assert 'prox' not in net.gid_ranges
+    assert net._n_gids == n_gids
+
+
 def test_add_drives():
     """Test methods for adding drives to a Network."""
     hnn_core_root = op.dirname(hnn_core.__file__)