diff --git a/pykat/utilities/plotting/beamtrace.py b/pykat/utilities/plotting/beamtrace.py index 33e0f309078156ee88870e876790ec06db0461e6..b7e315885c01c803cf65769dc66efbe7e0637c42 100644 --- a/pykat/utilities/plotting/beamtrace.py +++ b/pykat/utilities/plotting/beamtrace.py @@ -12,6 +12,9 @@ def plot_beam_trace(_kat, from_node, to_node): components = kat.nodes.getComponentsBetween(from_node, to_node) + if len(components) == 0: + raise pkex.BasePyKatException('Could not find a direct path between nodes %s and %s'%(from_node, to_node)) + nodes = [] for cn in range(len(components)):