Skip to content

Commit 026500c

Browse files
committed
tests: fix flacky test_create_job_configs_labels_log_adaptor_call_method_under_length_limit
1 parent 4da309e commit 026500c

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

tests/unit/session/test_io_bigquery.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -109,14 +109,16 @@ def test_create_job_configs_labels_log_adaptor_call_method_under_length_limit():
109109
labels = io_bq.create_job_configs_labels(
110110
job_configs_labels=cur_labels, api_methods=api_methods
111111
)
112-
expected_dict = {
113-
"source": "bigquery-dataframes-temp",
114-
"bigframes-api": "dataframe-columns",
115-
"recent-bigframes-api-0": "dataframe-max",
116-
"recent-bigframes-api-1": "dataframe-head",
117-
"recent-bigframes-api-2": "dataframe-__init__",
118-
}
119-
assert labels == expected_dict
112+
assert "source" in labels
113+
assert labels["source"] == "bigquery-dataframes-temp"
114+
assert "bigframes-api" in labels
115+
assert labels["bigframes-api"] == "dataframe-columns"
116+
assert "recent-bigframes-api-0" in labels
117+
assert labels["recent-bigframes-api-0"] == "dataframe-max"
118+
assert "recent-bigframes-api-1" in labels
119+
assert labels["recent-bigframes-api-1"] == "dataframe-head"
120+
assert "recent-bigframes-api-2" in labels
121+
assert labels["recent-bigframes-api-2"] == "dataframe-__init__"
120122

121123

122124
def test_create_job_configs_labels_length_limit_met_and_labels_is_none():

0 commit comments

Comments
 (0)