added Range.offset(), closes #89

xlwings · Nov 25, 2014 · a59fd6a · a59fd6a
1 parent 208a048
commit a59fd6a
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 4 deletions.
diff --git a/xlwings/main.py b/xlwings/main.py
@@ -937,11 +937,11 @@ def color(self, color_or_rgb):
 
     def resize(self, row_size=None, column_size=None):
         """
-        Resizes the specified range.
+        Resizes the specified Range.
 
         Returns
         -------
-        Range : a Range object that represents the resized Range.
+        Range : Range object
         """
         if row_size:
             row2 = self.row1 + row_size - 1
@@ -952,7 +952,30 @@ def resize(self, row_size=None, column_size=None):
         else:
             col2 = self.col1
 
-        return Range((self.row1, self.col1),(row2, col2))
+        return Range((self.row1, self.col1), (row2, col2))
+
+    def offset(self, row_offset=None, column_offset=None):
+        """
+        Returns a Range object that represents a Range that's offset from the specified range.
+
+        Returns
+        -------
+        Range : Range object
+        """
+
+        if row_offset:
+            row1 = self.row1 + row_offset
+            row2 = self.row2 + row_offset
+        else:
+            row1, row2 = self.row1, self.row2
+
+        if column_offset:
+            col1 = self.col1 + column_offset
+            col2 = self.col2 + column_offset
+        else:
+            col1, col2 = self.col1, self.col2
+
+        return Range((row1, col1), (row2, col2))
 
 
 class Chart(object):

diff --git a/xlwings/tests/test_xlwings.py b/xlwings/tests/test_xlwings.py
@@ -669,9 +669,19 @@ def test_resize(self):
         r = Range('A1').resize(row_size=4)
         assert_equal(r.shape, (4, 1))
 
-        r = Range('A1').resize(column_size=5)
+        r = Range('A1:B4').resize(column_size=5)
         assert_equal(r.shape, (1, 5))
 
+    def test_offset(self):
+        o = Range('A1:B3').offset(3, 4)
+        assert_equal(o.get_address(), '$E$4:$F$6')
+
+        o = Range('A1:B3').offset(row_offset=3)
+        assert_equal(o.get_address(), '$A$4:$B$6')
+
+        o = Range('A1:B3').offset(column_offset=4)
+        assert_equal(o.get_address(), '$E$1:$F$3')
+
 
 class TestChart:
     def setUp(self):