udf_python.py

@snippet
def udf_inline(client, _):
    """Inline UDF."""
    from google.cloud.bigquery.job import UDFResource
    UDF_QUERY = ('SELECT upper_name FROM upperCaseName('
                 '[bigquery-public-data:usa_names.usa_1910_2013]) '
                 'WHERE state = "TX"')
    INLINE_UDF_CODE = """
        function upperCaseName(r, emit) {
            emit({upper_name: r.name.toUpperCase(),
                  state: r.state});
        }

        bigquery.defineFunction(
          'upperCaseName',
          ['name', 'state'],
          [{'name': 'upper_name', 'type': 'string'},
           {'name': 'state', 'type': 'string'}],
          upperCaseName
        );
    """
    LIMIT = 100
    LIMITED = '%s LIMIT %d' % (UDF_QUERY, LIMIT)
    TIMEOUT_MS = 1000

    query = client.run_sync_query(LIMITED)
    query.timeout_ms = TIMEOUT_MS
    query.udf_resources = [UDFResource("inlineCode", INLINE_UDF_CODE)]
    query.run()     # API Query

    assert query.complete
    assert len(query.rows) == LIMIT
    for row in query.rows:
        assert row['name'].isUpper()