Skip to content

Commit 68e3d69

Browse files
modify finicky test
1 parent 3a00ab5 commit 68e3d69

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/unit/session/test_io_bigquery.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff 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

172178
def test_add_and_trim_labels_length_limit_met():

0 commit comments

Comments
 (0)