Makes the only_distance paramter value equivalent to value given to f…

…unction when granularity paramter is also provided by user
bbs-ld · Dec 10, 2017 · 0f60d3d · 0f60d3d
1 parent 66ac193
commit 0f60d3d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/arrow/arrow.py b/arrow/arrow.py
@@ -806,7 +806,7 @@ def humanize(self, other=None, locale='en_us', only_distance=False, granularity=
 
             if(trunc(abs(delta)) != 1):
                 granularity += 's'
-            return locale.describe(granularity, delta, only_distance=False)
+            return locale.describe(granularity, delta, only_distance=only_distance)
     # math
 
     def __add__(self, other):

diff --git a/tests/arrow_tests.py b/tests/arrow_tests.py
@@ -1166,6 +1166,10 @@ def test_granularity(self):
         later108 = self.now.shift(seconds=10 ** 8)
         assertEqual(self.now.humanize(later108, granularity = 'year'), '3 years ago')
         assertEqual(later108.humanize(self.now, granularity = 'year'), 'in 3 years')
+
+        later108onlydistance = self.now.shift(seconds=10 ** 8)
+        assertEqual(self.now.humanize(later108onlydistance , only_distance=True, granularity = 'year'), '3 years')
+        assertEqual(later108onlydistance .humanize(self.now, only_distance=True, granularity = 'year'), '3 years')
         with assertRaises(AttributeError):
             self.now.humanize(later108, granularity = 'years')