rangeset: Range support operator "&"

Author: drmingdrmer

rangeset/README.md

@@ -10,6 +10,7 @@
   - [rangeset.IntIncRange](#rangesetintincrange)
   - [rangeset.IntIncRangeSet](#rangesetintincrangeset)
   - [rangeset.Range](#rangesetrange)
+    - [Range `&`](#range-&)
     - [Range.cmp](#rangecmp)
     - [Range.has(val)](#rangehasval)
     - [Range.is_adjacent](#rangeis_adjacent)
@@ -99,6 +100,23 @@ E.g. a range `[1, 3]` has 2 elements `1` and `2`, but `3` is not in this range.
 **return**:
 a `rangeset.Range` instance.
 
+
+### Range `&`
+
+Overriding operator `&`.
+`a & b` is the same as `a.intersect(b)`.
+
+**syntax**:
+`Range(1, 3) & Range(2, 4) # Range(2, 3)`.
+
+**syntax**:
+`Range(1, 3) & [2, 4] # Range(2, 3)`.
+
+**syntax**:
+`[1, 3] & Range(2, 4) # Range(2, 3)`.
+
+
+
 ### Range.cmp
 
 **syntax**:

rangeset/rangeset.py

@@ -159,6 +159,11 @@ def val(self):
     def set(self, v):
         self[2] = v
 
+    def __and__(self, b):
+        return self.intersect(b)
+
+    def __rand__(self, b):
+        return self.intersect(b)
 
 class Range(ValueRange):
 

rangeset/test/test_rangeset.py

@@ -191,13 +191,29 @@ def test_intersect(self):
             self.assertEqual(expected, a.intersect(rangeset.Range(*b)))
             self.assertEqual(expected, a.intersect(rangeset.ValueRange(*(b+['bar']))))
 
+            # __and__
+            self.assertEqual(expected, a & b)
+            self.assertEqual(expected, a & (b+['bar']))
+
+            # __rand__
+            self.assertEqual(expected, b & a)
+            self.assertEqual(expected, (b+['bar']) & a)
+
             a = rangeset.ValueRange(*(a+['ahh']))
             if expected is not None:
                 expected = expected + ['ahh']
 
             self.assertEqual(expected, a.intersect(rangeset.Range(*b)))
             self.assertEqual(expected, a.intersect(rangeset.ValueRange(*(b+['bar']))))
 
+            # __and__
+            self.assertEqual(expected, a & b)
+            self.assertEqual(expected, a & (b+['bar']))
+
+            # __rand__
+            self.assertEqual(expected, b & a)
+            self.assertEqual(expected, (b+['bar']) & a)
+
     def test_length(self):
         inf = float('inf')
         cases = (