File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -158,15 +158,21 @@ def test_create_job_configs_labels_length_limit_met():
158158 df .max ()
159159 api_methods = log_adapter ._api_methods
160160
161+ assert set (api_methods ) == {
162+ "dataframe-max" ,
163+ "dataframe-head" ,
164+ "dataframe-__init__" ,
165+ "session-__init__" ,
166+ }
161167 labels = io_bq .create_job_configs_labels (
162168 job_configs_labels = cur_labels , api_methods = api_methods
163169 )
164170 assert labels is not None
165- assert len (labels ) == 56
166171 assert "dataframe-max" in labels .values ()
167172 assert "dataframe-head" not in labels .values ()
168173 assert "bigframes-api" in labels .keys ()
169174 assert "source" in labels .keys ()
175+ assert len (labels ) == 56
170176
171177
172178def test_add_and_trim_labels_length_limit_met ():
You can’t perform that action at this time.
0 commit comments