Fix test of 'ItemsView.__contains__' magic method

lycantropos · Mar 25, 2023 · 9472a88 · 9472a88
1 parent e9ff6c7
commit 9472a88
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/views_tests/items_view_tests/test_contains.py b/tests/views_tests/items_view_tests/test_contains.py
@@ -5,6 +5,7 @@
 from dendroid.hints import Item
 from tests.utils import (ItemsView,
                          are_items_equal,
+                         are_items_keys_equal,
                          equivalence,
                          to_items_view_including_item)
 from . import strategies
@@ -18,11 +19,14 @@ def test_base_case(items_view_with_item: Tuple[ItemsView, Item]) -> None:
 
 
 @given(strategies.items_views_with_items_pairs)
-def test_step(items_view_with_items_pair: Tuple[ItemsView, Tuple[Item, Item]]
-              ) -> None:
+def test_step(
+        items_view_with_items_pair: Tuple[ItemsView, Tuple[Item, Item]]
+) -> None:
     items_view, (extra_item, item) = items_view_with_items_pair
 
     next_items_view = to_items_view_including_item(items_view, extra_item)
 
     assert equivalence(item in next_items_view,
-                       item in items_view or are_items_equal(item, extra_item))
+                       (item in items_view
+                        and not are_items_keys_equal(item, extra_item))
+                       or are_items_equal(item, extra_item))