From 19ecb8d5e8fa5d09821d6f7a08d747d427758027 Mon Sep 17 00:00:00 2001 From: Joe Jevnik Date: Mon, 1 Aug 2016 14:25:36 -0400 Subject: [PATCH] TST: add tests for range.__eq__ --- zipline/utils/range.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/zipline/utils/range.py b/zipline/utils/range.py index 0e2f46e1..138b0922 100644 --- a/zipline/utils/range.py +++ b/zipline/utils/range.py @@ -1,4 +1,3 @@ -from functools import reduce import operator as op from six import PY2 @@ -70,7 +69,22 @@ if PY2: return hash((type(self), self.start, self.stop, self.step)) def __eq__(self, other): - return reduce( + """ + Examples + -------- + >>> range(1) == range(1) + True + >>> range(0, 5, 2) == range(0, 5, 2) + True + >>> range(5, 0, -2) == range(5, 0, -2) + True + + >>> range(1) == range(2) + False + >>> range(0, 5, 2) == range(0, 5, 3) + False + """ + return all( getattr(self, attr) == getattr(other, attr) for attr in self.__slots__ )