overloaded getitem on transport to return empty dict if pore.source i…

…s requested
AlemSnyder · Nov 5, 2022 · 31f6b26 · 31f6b26
1 parent d1526eb
commit 31f6b26
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/openpnm/algorithms/_transport.py b/openpnm/algorithms/_transport.py
@@ -73,6 +73,15 @@ def __init__(self, phase, name='trans_?', **kwargs):
         self._pure_b = None
         self.soln = {}
 
+    def __getitem__(self, key):
+        try:
+            return super().__getitem__(key)
+        except KeyError:
+            if key == 'pore.source':
+                return {}
+            else:
+                raise KeyError(key)
+
     @property
     def x(self):
         """Shortcut to the solution currently stored on the algorithm."""

diff --git a/tests/unit/algorithms/GenericTransportTest.py b/tests/unit/algorithms/GenericTransportTest.py
@@ -357,7 +357,7 @@ def test_rate_multiple_values(self):
 
     def test_x0_is_nan(self):
         alg = op.algorithms.Transport(network=self.net,
-                                             phase=self.phase)
+                                      phase=self.phase)
         alg.settings['conductance'] = 'throat.diffusive_conductance'
         alg.settings['quantity'] = 'pore.mole_fraction'
         alg.set_value_BC(pores=self.net.pores('top'), values=1)
@@ -367,6 +367,13 @@ def test_x0_is_nan(self):
         with pytest.raises(Exception):
             alg.run(x0=x0)
 
+    def test_get_source_list(self):
+        alg = op.algorithms.Transport(network=self.net,
+                                      phase=self.phase)
+        assert alg['pore.source'] == {}
+        with pytest.raises(KeyError):
+            alg['pore.source_blah'] == {}
+
     def teardown_class(self):
         ws = op.Workspace()
         ws.clear()