@@ -139,42 +139,6 @@ def test_create_job_configs_labels_length_limit_met_and_labels_is_none():
139139 assert "dataframe-head" in labels .values ()
140140
141141
142- def test_create_job_configs_labels_length_limit_met ():
143- log_adapter .get_and_reset_api_methods ()
144- cur_labels = {
145- "bigframes-api" : "read_pandas" ,
146- "source" : "bigquery-dataframes-temp" ,
147- }
148- for i in range (53 ):
149- key = f"bigframes-api-test-{ i } "
150- value = f"test{ i } "
151- cur_labels [key ] = value
152- # If cur_labels length is 62, we can only add one label from api_methods
153- df = bpd .DataFrame (
154- {"col1" : [1 , 2 ], "col2" : [3 , 4 ]}, session = mocks .create_bigquery_session ()
155- )
156- # Test running two methods
157- df .head ()
158- df .max ()
159- api_methods = log_adapter ._api_methods
160-
161- assert set (api_methods ) == {
162- "dataframe-max" ,
163- "dataframe-head" ,
164- "dataframe-__init__" ,
165- "session-__init__" ,
166- }
167- labels = io_bq .create_job_configs_labels (
168- job_configs_labels = cur_labels , api_methods = api_methods
169- )
170- assert labels is not None
171- assert "dataframe-max" in labels .values ()
172- assert "dataframe-head" not in labels .values ()
173- assert "bigframes-api" in labels .keys ()
174- assert "source" in labels .keys ()
175- assert len (labels ) == 56
176-
177-
178142def test_add_and_trim_labels_length_limit_met ():
179143 log_adapter .get_and_reset_api_methods ()
180144 cur_labels = {
0 commit comments