Add Set.__iter__()

kojiishi · Apr 6, 2024 · edd336c · edd336c
1 parent c136e16
commit edd336c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tests/set_test.py b/tests/set_test.py
@@ -16,6 +16,13 @@ def test_set_add_remove():
     s.remove(3)
 
 
+def test_set_iter():
+    s = ur.Set()
+    s.add(1)
+    s.add(5)
+    assert list(s) == [1, 5]
+
+
 def test_set_general_category():
     reader = ur.UnicodeDataReader()
     l = ur.Set.general_category('L', reader)

diff --git a/unicodedata_reader/set.py b/unicodedata_reader/set.py
@@ -13,6 +13,9 @@ def __init__(self) -> None:
     def __contains__(self, code_point: int) -> bool:
         return code_point in self.set
 
+    def __iter__(self) -> Iterable[int]:
+        return self.set.__iter__()
+
     def __isub__(self, other: 'Set') -> None:
         self.set -= other.set