Revised the method to reset node types.

src/topupopt/problems/esipp/network.py

@@ -752,9 +752,12 @@ class Network(nx.MultiDiGraph):
         elif self.is_source_sink_node(node_key):
             # source/sink node
             self.source_sink_nodes.remove(node_key)
-        else: # self.is_waypoint_node(node_key):
+        elif self.is_waypoint_node(node_key):
             # has to be a waypoint node
             self.waypoint_nodes.remove(node_key)
+        # No need to reset node but this could mean something is up
+        # else:
+        #     raise ValueError('Unknown node type.')
 
     # *************************************************************************
     # *************************************************************************

tests/test_dhn_utils.py

@@ -20,7 +20,6 @@ from topupheat.common.fluids import FluidDatabase  # , Fluid
 # *****************************************************************************
 # *****************************************************************************
 
-
 class TestDistrictHeatingNetworkUtils:
     # *************************************************************************
     # *************************************************************************