Class ImpalaTestVector

tests/common/test_vector.py:92–145

# A test vector that passed to test method. The ImpalaTestVector can be used to
# extract values for the specified dimension(s)
class ImpalaTestVector(object):
  def __init__(self, vector_values):
    self.vector_values = vector_values

  def get_value_with_default(self, name, default_value):
    for vector_value in self.vector_values:
      if vector_value.name == name:
        return vector_value.value
    return default_value

  def get_value(self, name):
    for vector_value in self.vector_values:
      if vector_value.name == name:
        return vector_value.value
    raise ValueError("Test vector does not contain value '%s'" % name)

  def get_protocol(self):
    return self.get_value(PROTOCOL)

  def get_table_format(self):
    return self.get_value(TABLE_FORMAT)

  def get_exec_option_dict(self):
    return self.get_value(EXEC_OPTION)

  def get_exec_option(self, option_name):
    value = self.get_value(EXEC_OPTION).get(option_name.lower())
    assert value is not None
    return value

  def set_exec_option(self, option_name, option_value):
    self.get_value(EXEC_OPTION)[option_name.lower()] = str(option_value)

  def unset_exec_option(self, option_name):
    del self.get_value(EXEC_OPTION)[option_name.lower()]

  def __str__(self):
      return ' | '.join(['%s' % vector_value for vector_value in self.vector_values])

  def __repr__(self):
    return str(self)

  # Each value in a test vector is wrapped in the Value object. This wrapping is
  # done internally so this object should never need to be created by the user.
  class Value(object):
    def __init__(self, name, value):
      self.name = name
      self.value = value

    def __str__(self):
      return '"%s: %s"' % (self.name, self.value)

    def __repr__(self):
      return str(self)


# Matrix -> Collection of vectors