fix test_load_table_from_dataframe_w_higher_scale_decimal128_datatype

Author: Gaurang033

google/cloud/bigquery/_pandas_helpers.py

@@ -23,6 +23,7 @@
 import warnings
 from typing import Any, Union
 
+
 from google.cloud.bigquery import _pyarrow_helpers
 from google.cloud.bigquery import _versions_helpers
 from google.cloud.bigquery import schema
@@ -485,7 +486,6 @@ def augment_schema(dataframe, current_bq_schema):
     # pytype: disable=attribute-error
     augmented_schema = []
     unknown_type_fields = []
-
     for field in current_bq_schema:
         if field.field_type is not None:
             augmented_schema.append(field)
@@ -515,9 +515,7 @@ def augment_schema(dataframe, current_bq_schema):
         else:
             detected_mode = field.mode
             detected_type = _pyarrow_helpers.arrow_scalar_ids_to_bq(arrow_table.type.id)
-            if detected_type == "NUMERIC" and (
-                arrow_table.type.precision > 38 or arrow_table.type.scale > 9
-            ):
+            if detected_type == "NUMERIC" and arrow_table.type.scale > 9:
                 detected_type = "BIGNUMERIC"
 
         if detected_type is None:

tests/unit/test_client.py

@@ -8900,24 +8900,14 @@ def test_load_table_from_dataframe_w_higher_scale_decimal128_datatype(self):
         from decimal import Decimal
 
         client = self._make_client()
-        dataframe = pandas.DataFrame(
-            {
-                "x": [
-                    Decimal("0.12345678901234560000000000000000000000"),
-                    Decimal("01234567890123456789012345678901234567.1234567891"),
-                ]
-            }
-        )
+        dataframe = pandas.DataFrame({"x": [Decimal("0.1234567891")]})
         load_patch = mock.patch(
             "google.cloud.bigquery.client.Client.load_table_from_file", autospec=True
         )
 
         get_table_patch = mock.patch(
-            "google.cloud.bigquery.client.Client.get_table",
-            autospec=True,
-            return_value=mock.Mock(schema=[SchemaField("x", "BIGNUMERIC", "NULLABLE")]),
+            "google.cloud.bigquery.client.Client.get_table", autospec=True
         )
-
         with load_patch as load_table_from_file, get_table_patch:
             client.load_table_from_dataframe(
                 dataframe, self.TABLE_REF, location=self.LOCATION