loadBigQueryTable() and runBigQuery(): Add static methods and unit tests.

PiperOrigin-RevId: 740773370

Author: Filip Domański

python/ee/featurecollection.py

@@ -331,6 +331,25 @@ def kriging(
         reducer,
     )
 
+  @staticmethod
+  def loadBigQueryTable(
+      table: _arg_types.String,
+      geometryColumn: Optional[_arg_types.String] = None,
+  ) -> FeatureCollection:
+    """Returns a FeatureCollection containing data read from a BigQuery table.
+
+    Args:
+      table: Path to BigQuery table in a project.dataset.table format.
+      geometryColumn: The name of the column to use as the main feature
+        geometry. If not specified, all features will have null geometry.
+    """
+
+    return apifunction.ApiFunction.call_(
+        'FeatureCollection.loadBigQueryTable',
+        table,
+        geometryColumn,
+    )
+
   def makeArray(
       self,
       properties: _arg_types.List,
@@ -384,3 +403,27 @@ def randomPoints(
     return apifunction.ApiFunction.call_(
         'FeatureCollection.randomPoints', region, points, seed, maxError
     )
+
+  @staticmethod
+  def runBigQuery(
+      query: _arg_types.String,
+      geometryColumn: Optional[_arg_types.String] = None,
+      maxBytesBilled: Optional[_arg_types.Integer] = int(1e11),
+  ) -> FeatureCollection:
+    """Returns a FeatureCollection containing result of a BigQuery query.
+
+    Args:
+      query: GoogleSQL query to perform on the BigQuery resources.
+      geometryColumn: The name of the column to use as the main feature
+        geometry. If not specified, all features will have null geometry.
+      maxBytesBilled: Maximum number of bytes billed while processing the query.
+        Any BigQuery job that exceeds this limit will fail and won't be billed.
+        Defaults to 100GB.
+    """
+
+    return apifunction.ApiFunction.call_(
+        'FeatureCollection.runBigQuery',
+        query,
+        geometryColumn,
+        maxBytesBilled
+    )

python/ee/tests/featurecollection_test.py

@@ -770,6 +770,26 @@ def test_kriging(self):
     result = json.loads(expression.serialize())
     self.assertEqual(expect, result)
 
+  def test_load_big_query_table(self):
+    table = 'bigquery-public-data.new_york_subway.stations'
+    geometry_column = 'geometry'
+    expect = make_expression_graph({
+        'arguments': {
+            'table': {'constantValue': table},
+            'geometryColumn': {'constantValue': geometry_column},
+        },
+        'functionName': 'FeatureCollection.loadBigQueryTable',
+    })
+    expression = ee.FeatureCollection.loadBigQueryTable(table, geometry_column)
+    result = json.loads(expression.serialize())
+    self.assertEqual(expect, result)
+
+    expression = ee.FeatureCollection.loadBigQueryTable(
+        table=table, geometryColumn=geometry_column
+    )
+    result = json.loads(expression.serialize())
+    self.assertEqual(expect, result)
+
   def test_make_array(self):
     properties = ['a', 'b']
     name = 'name string'
@@ -951,7 +971,32 @@ def test_remap(self):
     self.assertEqual(expect, result)
 
     expression = collection.remap(
-        lookupIn=lookup_in, lookupOut=lookup_out, columnName=column_name
+        lookupIn=lookup_in, lookupOut=lookup_out, columnName=column_name)
+    result = json.loads(expression.serialize())
+    self.assertEqual(expect, result)
+
+  def test_run_big_query(self):
+    query = 'SELECT * FROM `bigquery-public-data.new_york_subway.stations`'
+    geometry_column = 'geometry'
+    max_bytes_billed = 1000
+    expect = make_expression_graph({
+        'arguments': {
+            'query': {'constantValue': query},
+            'geometryColumn': {'constantValue': geometry_column},
+            'maxBytesBilled': {'constantValue': max_bytes_billed},
+        },
+        'functionName': 'FeatureCollection.runBigQuery',
+    })
+    expression = ee.FeatureCollection.runBigQuery(
+        query, geometry_column, max_bytes_billed
+    )
+    result = json.loads(expression.serialize())
+    self.assertEqual(expect, result)
+
+    expression = ee.FeatureCollection.runBigQuery(
+        query=query,
+        geometryColumn=geometry_column,
+        maxBytesBilled=max_bytes_billed,
     )
     result = json.loads(expression.serialize())
     self.assertEqual(expect, result)