From 20decfa918679ce23fb444dd840e6f5f74e8e862 Mon Sep 17 00:00:00 2001 From: Chris Lenk Date: Thu, 8 Feb 2024 07:50:18 -0500 Subject: [PATCH] Fix atomic planner tests Use assert_called_with (https://github.com/python/cpython/issues/100690) --- tests/planners/test_atomic.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/planners/test_atomic.py b/tests/planners/test_atomic.py index 11f3a67e8..bd481fa69 100644 --- a/tests/planners/test_atomic.py +++ b/tests/planners/test_atomic.py @@ -35,6 +35,9 @@ class LinkStub(): def __init__(self, ability_id): self.ability = AbilityStub(ability_id) + def __eq__(self, other): + return self.ability.ability_id == other.ability.ability_id + @pytest.fixture def atomic_planner(): @@ -61,8 +64,8 @@ def test_atomic_with_links_in_order(self, event_loop, atomic_planner): assert atomic_planner.operation.apply.call_count == 1 assert atomic_planner.operation.wait_for_links_completion.call_count == 1 - assert atomic_planner.operation.apply.called_with(LinkStub('ability_b')) - assert atomic_planner.operation.wait_for_links_completion.called_with([LinkStub('ability_b')]) + atomic_planner.operation.apply.assert_called_with(LinkStub('ability_b')) + atomic_planner.operation.wait_for_links_completion.assert_called_with([LinkStub('ability_b')]) def test_atomic_with_links_out_of_order(self, event_loop, atomic_planner): @@ -77,8 +80,8 @@ def test_atomic_with_links_out_of_order(self, event_loop, atomic_planner): assert atomic_planner.operation.apply.call_count == 1 assert atomic_planner.operation.wait_for_links_completion.call_count == 1 - assert atomic_planner.operation.apply.called_with(LinkStub('ability_b')) - assert atomic_planner.operation.wait_for_links_completion.called_with([LinkStub('ability_b')]) + atomic_planner.operation.apply.assert_called_with(LinkStub('ability_b')) + atomic_planner.operation.wait_for_links_completion.assert_called_with([LinkStub('ability_b')]) def test_atomic_no_links(self, event_loop, atomic_planner):