Function test_write_query

tests/comparison/tests/test_query_objects.py:84–93

(sql_writer, query_test)

                         ids=_idfn)
@pytest.mark.parametrize('sql_writer', ['IMPALA', 'POSTGRESQL'], indirect=True)
def test_write_query(sql_writer, query_test):
  if isinstance(sql_writer, ImpalaSqlWriter):
    expected_string = getattr(query_test, 'impala_query_string')
  elif isinstance(sql_writer, PostgresqlSqlWriter):
    expected_string = getattr(query_test, 'postgres_query_string')
  else:
    raise Exception('unsupported writer: {0}'.format(sql_writer))
  verify_sql_matches(
      sql_writer.write_query(query_test.query),
      expected_string)