Skip to content

Commit

Permalink
Fixed tags comparesion, as it's now a set, and not a list.
Browse files Browse the repository at this point in the history
  • Loading branch information
Avihais12344 committed Sep 20, 2024
1 parent bae6471 commit 18773a7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/models/test_dagbag.py
Original file line number Diff line number Diff line change
Expand Up @@ -830,7 +830,7 @@ def test_get_dag_with_dag_serialization(self):
# from DB
with time_machine.travel((tz.datetime(2020, 1, 5, 0, 0, 4)), tick=False):
with assert_queries_count(0):
assert dag_bag.get_dag("example_bash_operator").tags == ["example", "example2"]
assert dag_bag.get_dag("example_bash_operator").tags == {"example", "example2"}

# Make a change in the DAG and write Serialized DAG to the DB
with time_machine.travel((tz.datetime(2020, 1, 5, 0, 0, 6)), tick=False):
Expand Down Expand Up @@ -869,7 +869,7 @@ def test_get_dag_refresh_race_condition(self):
ser_dag = dag_bag.get_dag("example_bash_operator")

ser_dag_update_time = dag_bag.dags_last_fetched["example_bash_operator"]
assert ser_dag.tags == ["example", "example2"]
assert ser_dag.tags == {"example", "example2"}
assert ser_dag_update_time == tz.datetime(2020, 1, 5, 1, 0, 10)

with create_session() as session:
Expand Down

0 comments on commit 18773a7

Please sign in to comment.