Utility for writing multi-host tests for pytest