docs: def wait(self, condition):
          """
          Waits for a condition to be true.
  
          :param condition: Should be a callable that operates in the
                            same way ``WebDriverWait.until`` expects.
          :returns: Whatever ``WebDriverWait.until`` returns.
          """
          return WebDriverWait(self.driver, self.timeout).until(condition)