[PortList.measure_travel] add a convenience wrapper for measuring internal travel

This commit is contained in:
Jan Petykiewicz 2026-04-09 11:41:21 -07:00
commit 6cf9600193
2 changed files with 29 additions and 0 deletions

View file

@ -273,6 +273,12 @@ class PortList(metaclass=ABCMeta):
else: # noqa: RET505
return {k: self.ports[k] for k in key}
def measure_travel(self, src: str, dst: str) -> tuple[NDArray[numpy.float64], float | None]:
"""
Convenience wrapper for measuring travel between two named ports.
"""
return self[src].measure_travel(self[dst])
def __contains__(self, key: str) -> NoReturn:
raise NotImplementedError('PortsList.__contains__ is left unimplemented. Use `key in container.ports` instead.')