Merge pull request #1197 from alphagov/insensitive-set-contains

Implement `InsensitiveSet.__contains__`
alphagov · Feb 26, 2025 · a6b6dd1 · a6b6dd1
2 parents 1b1d0ef + b1da482
commit a6b6dd1
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## 95.2.0
+
+* Implement `InsensitiveSet.__contains__`
+
 ## 95.1.2
 
 * Fix to the number of arguments the `on_retry` of the `NotifyTask` class takes.

diff --git a/notifications_utils/insensitive_dict.py b/notifications_utils/insensitive_dict.py
@@ -63,3 +63,6 @@ def make_key(original_key):
 class InsensitiveSet(OrderedSet):
     def __init__(self, iterable):
         return super().__init__(InsensitiveDict.from_keys(iterable).values())
+
+    def __contains__(self, key):
+        return key in InsensitiveDict.from_keys(self)
diff --git a/notifications_utils/version.py b/notifications_utils/version.py
@@ -5,4 +5,4 @@
 # - `make version-minor` for new features
 # - `make version-patch` for bug fixes
 
-__version__ = "95.1.2"  # a47171c8774fee6d72573d446710b651
+__version__ = "95.2.0"  # e423953f9dec74010897c62aa4e035de
diff --git a/tests/test_insensitive_dict.py b/tests/test_insensitive_dict.py
@@ -112,6 +112,27 @@ def test_insensitive_set():
     )
 
 
+def test_insensitive_set_contains():
+    foobar = InsensitiveSet(("foo", "bar"))
+
+    for key in (
+        "foo",
+        "F o o ",
+        "F_O_O",
+        "B_A_R",
+        "B a r",
+        "bar",
+    ):
+        assert key in foobar
+
+    for key in (
+        "baz",
+        "barz",
+        "z foo",
+    ):
+        assert key not in foobar
+
+
 @pytest.mark.parametrize(
     "extra_args, expected_dict",
     (