fix #40

bluepysnap/circuit.py

@@ -70,3 +70,12 @@ def edges(self):
             self._config['networks']['edges'],
             lambda cfg: EdgeStorage(cfg, self)
         )
+
+    def close_contexts(self):
+        """Close the context for all populations."""
+        if self.nodes:
+            for population in self.nodes.values():
+                population.close_context()
+        if self.edges:
+            for population in self.edges.values():
+                population.close_context()

bluepysnap/edges.py

@@ -50,7 +50,7 @@ def __init__(self, config, circuit):
         self._circuit = circuit
         self._populations = {}
 
-    @cached_property
+    @property
     def storage(self):
         """Access to the libsonata edge storage."""
         return libsonata.EdgeStorage(self._h5_filepath)
@@ -113,7 +113,12 @@ def __init__(self, edge_storage, population_name):
     def _population(self):
         return self._edge_storage.storage.open_population(self.name)
 
-    @property
+    def close_context(self):
+        """Close the h5 context for edge population."""
+        if "_population" in self.__dict__:
+            del self.__dict__["_population"]
+
+    @cached_property
     def size(self):
         """Population size."""
         return self._population.size
@@ -396,7 +401,6 @@ def _optimal_direction():
             secondary_node_ids = np.unique(secondary_node_ids)
 
         secondary_node_ids_used = set()
-
         for key_node_id in primary_node_ids:
             connected_node_ids = get_connected_node_ids(key_node_id, unique=False)
             connected_node_ids_with_count = np.stack(

bluepysnap/nodes.py

@@ -54,7 +54,7 @@ def __init__(self, config, circuit):
         self._circuit = circuit
         self._populations = {}
 
-    @cached_property
+    @property
     def storage(self):
         """Access to the libsonata node storage."""
         return libsonata.NodeStorage(self._h5_filepath)
@@ -176,6 +176,11 @@ def _data(self):
     def _population(self):
         return self._node_storage.storage.open_population(self.name)
 
+    def close_context(self):
+        """Close the h5 context for node population."""
+        if "_population" in self.__dict__:
+            del self.__dict__["_population"]
+
     @cached_property
     def size(self):
         """Node population size."""

tests/test_circuit.py

@@ -1,4 +1,5 @@
 import pytest
+import h5py
 
 from bluepysnap.nodes import NodePopulation
 from bluepysnap.edges import EdgePopulation
@@ -11,12 +12,12 @@
 def test_all():
     circuit = test_module.Circuit(
         str(TEST_DATA_DIR / 'circuit_config.json'))
-    assert(
-        circuit.config['networks']['nodes'][0] ==
-        {
-            'nodes_file': str(TEST_DATA_DIR / 'nodes.h5'),
-            'node_types_file': None,
-        }
+    assert (
+            circuit.config['networks']['nodes'][0] ==
+            {
+                'nodes_file': str(TEST_DATA_DIR / 'nodes.h5'),
+                'node_types_file': None,
+            }
     )
     assert isinstance(circuit.nodes, dict)
     assert isinstance(circuit.edges, dict)
@@ -34,3 +35,20 @@ def test_duplicate_population():
     with pytest.raises(BluepySnapError):
         circuit.nodes
 
+
+def test_close_contexts():
+    circuit = test_module.Circuit(
+        str(TEST_DATA_DIR / 'circuit_config.json')
+    )
+    circuit.edges['default'].size
+    circuit.nodes['default'].size
+    node_file = circuit.config['networks']['nodes'][0]['nodes_file']
+    edge_file = circuit.config['networks']['edges'][0]['edges_file']
+
+    circuit.close_contexts()
+
+    with h5py.File(node_file, "r+") as h5:
+        list(h5)
+
+    with h5py.File(edge_file, "r+") as h5:
+        list(h5)

tests/test_edges.py

@@ -4,6 +4,7 @@
 import pandas as pd
 import pandas.testing as pdt
 import pytest
+import h5py
 
 import libsonata
 from mock import Mock

tests/test_nodes.py

@@ -1,8 +1,9 @@
 import numpy as np
 import numpy.testing as npt
 import pandas as pd
-import pandas.util.testing as pdt
+import pandas.testing as pdt
 import pytest
+import h5py
 
 import libsonata
 from mock import Mock