Sorted keys and items views set-operations return sorted sets

Author: grantjenks

sortedcontainers/sorteddict.py

@@ -19,6 +19,7 @@
 from collections import ItemsView, KeysView, ValuesView, Sequence
 
 from .sortedlist import SortedList, recursive_repr
+from .sortedset import SortedSet
 
 
 class SortedDict(dict):
@@ -554,7 +555,7 @@ def _view_delitem(self, index):
 
     ``view.__delitem__(index)`` <==> ``del view[index]``
 
-    Support slicing.
+    Supports slicing.
 
     Runtime complexity: `O(log(n))` -- approximate.
 
@@ -598,6 +599,11 @@ class SortedKeysView(KeysView, Sequence):
     __slots__ = ()
 
 
+    @classmethod
+    def _from_iterable(self, it):
+        return SortedSet(it)
+
+
     def __getitem__(self, index):
         """Lookup key at `index` in sorted keys views.
 
@@ -642,6 +648,11 @@ class SortedItemsView(ItemsView, Sequence):
     __slots__ = ()
 
 
+    @classmethod
+    def _from_iterable(self, it):
+        return SortedSet(it)
+
+
     def __getitem__(self, index):
         """Lookup item at `index` in sorted items view.
 

tests/test_coverage_sorteddict.py

@@ -377,10 +377,10 @@ def test_keysview():
     assert keys <= that_keys
     assert keys >= that_keys
 
-    assert sorted(keys & that_keys) == [val for val, pos in mapping]
-    assert sorted(keys | that_keys) == [val for val, pos in mapping]
-    assert sorted(keys - that_keys) == []
-    assert sorted(keys ^ that_keys) == []
+    assert list(keys & that_keys) == [val for val, pos in mapping]
+    assert list(keys | that_keys) == [val for val, pos in mapping]
+    assert list(keys - that_keys) == []
+    assert list(keys ^ that_keys) == []
 
     keys = SortedDict(mapping[:2]).keys()
     assert repr(keys) == "SortedKeysView(SortedDict({'a': 0, 'b': 1}))"
@@ -447,10 +447,10 @@ def test_itemsview():
     assert items <= that_items
     assert items >= that_items
 
-    assert sorted(items & that_items) == mapping
-    assert sorted(items | that_items) == mapping
-    assert sorted(items - that_items) == []
-    assert sorted(items ^ that_items) == []
+    assert list(items & that_items) == mapping
+    assert list(items | that_items) == mapping
+    assert list(items - that_items) == []
+    assert list(items ^ that_items) == []
 
     items = SortedDict(mapping[:2]).items()
     assert repr(items) == "SortedItemsView(SortedDict({'a': 0, 'b': 1}))"