Skip to content

Commit

Permalink
Moved tests for iter() and reversed() on DiscreteRange to its proper …
Browse files Browse the repository at this point in the history
…location
  • Loading branch information
runfalk committed Jun 8, 2017
1 parent a74da15 commit fe286bf
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 35 deletions.
33 changes: 24 additions & 9 deletions tests/test_discreterange.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,27 @@ def test_last():
assert intrange(1).last is None


@pytest.mark.parametrize("a, b", [
(intrange(1, 10), range(1, 10)),
(
daterange(date(2000, 1, 1), date(2000, 1, 10)),
(date(2000, 1, i) for i in range(1, 10))
),
])
def test_iteration(a, b):
assert list(a) == list(b)
def test_iter():
assert list(intrange(0, 5)) == list(range(5))

infinite_iter = iter(intrange(0))
for i in range(100):
assert i == next(infinite_iter)


def test_no_lower_bound_iter():
with pytest.raises(TypeError):
next(iter(intrange(upper=1)))


def test_reversed():
assert list(reversed(intrange(0, 5))) == list(reversed(range(5)))

infinite_iter = reversed(intrange(upper=100))
for i in reversed(range(100)):
assert i == next(infinite_iter)


def test_no_lower_upper_reversed():
with pytest.raises(TypeError):
next(reversed(intrange(1)))
26 changes: 0 additions & 26 deletions tests/test_intrange.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,29 +4,3 @@

def test_len():
assert len(intrange(0, 5)) == 5


def test_iter():
assert list(intrange(0, 5)) == list(range(5))

infinite_iter = iter(intrange(0))
for i in range(100):
assert i == next(infinite_iter)


def test_no_lower_bound_iter():
with pytest.raises(TypeError):
next(iter(intrange(upper=1)))


def test_reversed():
assert list(reversed(intrange(0, 5))) == list(reversed(range(5)))

infinite_iter = reversed(intrange(upper=100))
for i in reversed(range(100)):
assert i == next(infinite_iter)


def test_no_lower_upper_reversed():
with pytest.raises(TypeError):
next(reversed(intrange(1)))

0 comments on commit fe286bf

Please sign in to comment.