Remove dead code and associated test.

Author: ericfe-google

bigquery_magics/bigquery.py

@@ -634,44 +634,6 @@ def _colab_node_expansion_callback(request: dict, params_str: str):
 MAX_GRAPH_VISUALIZATION_QUERY_RESULT_SIZE = 100_000
 
 
-def _estimate_json_size(df: pandas.DataFrame) -> int:
-    """Approximates the length of df.to_json(orient='records')
-    without materializing the string.
-    """
-    num_rows, num_cols = df.shape
-    if num_rows == 0:
-        return 2  # "[]"
-
-    # 1. Key overhead: "column_name": (repeated for every row)
-    # Includes quotes, colon, and comma separator per field
-    key_overhead = sum(len(f'"{col}":') + 1 for col in df.columns) * num_rows
-
-    # 2. Row structural overhead: { } per row and [ ] for the list
-    # Plus commas between rows (num_rows - 1)
-    structural_overhead = (2 * num_rows) + 2 + (num_rows - 1)
-
-    # 3. Value lengths
-    total_val_len = 0
-    for col in df.columns:
-        series = df[col]
-
-        if pandas.api.types.is_bool_dtype(series):
-            # true (4) or false (5)
-            total_val_len += series.map({True: 4, False: 5}).sum()
-        elif pandas.api.types.is_numeric_dtype(series):
-            # Numeric values (no quotes). Sample for average length to save memory.
-            sample_size = min(len(series), 1000)
-            avg_len = series.sample(sample_size).astype(str).str.len().mean()
-            total_val_len += avg_len * num_rows
-        else:
-            # Strings/Objects: "value" + quotes (2) + rough escaping factor
-            # .str.len() is relatively memory-efficient
-            val_chars = series.astype(str).str.len().sum()
-            total_val_len += val_chars + (2 * num_rows)
-
-    return int(key_overhead + structural_overhead + total_val_len)
-
-
 def _get_graph_name(query_text: str):
     """Returns the name of the graph queried.
 

tests/unit/bigquery/test_bigquery.py

@@ -3186,35 +3186,3 @@ def test_test_bigquery_magic__extension_loaded__is_registered_set_to_true():
     ip.extension_manager.load_extension("bigquery_magics")
 
     assert bigquery_magics.is_registered is True
-
-
-def test__estimate_json_size_empty():
-    df = pandas.DataFrame()
-    assert magics._estimate_json_size(df) == 2
-
-
-def test__estimate_json_size_bool():
-    df = pandas.DataFrame({"a": [True, False]}, dtype="bool")
-    # key_overhead: (len('"a":') + 1) * 2 = 5 * 2 = 10
-    # structural_overhead: (2 * 2) + 2 + (2 - 1) = 7
-    # total_val_len: 4 (true) + 5 (false) = 9
-    # total = 10 + 7 + 9 = 26
-    assert magics._estimate_json_size(df) == 26
-
-
-def test__estimate_json_size_int():
-    df = pandas.DataFrame({"a": [3, 4]})
-    # key_overhead: (len('"a":') + 1) * 2 = 5 * 2 = 10
-    # structural_overhead: (2 * 2) + 2 + (2 - 1) = 7
-    # total_val_len: 4 (true) + 5 (false) = 9
-    # total = 10 + 7 + 9 = 26
-    assert magics._estimate_json_size(df) == 19
-
-
-def test__estimate_json_size_string():
-    df = pandas.DataFrame({"a": ["x", "yz"]})
-    # key_overhead: (len('"a":') + 1) * 2 = 5 * 2 = 10
-    # structural_overhead: (2 * 2) + 2 + (2 - 1) = 7
-    # total_val_len: (1 + 2) + (2 * 2) = 7
-    # total = 10 + 7 + 7 = 24
-    assert magics._estimate_json_size(df) == 24