Skip to content

Commit 21a1be5

Browse files
author
Tercio de Melo
committed
Makes /after/ lazy
1 parent 4df2bdb commit 21a1be5

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

syntax_sugar/infix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def __rtruediv__(self, left):
1919
ffilter = infix(flip(filter))
2020
freduce = infix(flip(reduce))
2121
take = infix(compose(list, islice))
22-
after = infix(compose(list, lambda seq, idx: islice(seq, idx, None)))
22+
after = infix(lambda seq, idx: islice(seq, idx, None))
2323

2424
INF = float('inf')
2525
NEGINF = float('-inf')

tests/test_infix.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ def test_take():
8787

8888
def test_after():
8989
assert list(1 /to/ 10 /after/ 2 /take/ 3) == [3, 4, 5]
90+
assert list(1 /to/ INF /after/ 2 /take/ 3) == [3, 4, 5]
9091
assert list(10 /to/ 1 /after/ 3 /take/ 2) == [7, 6]
9192

9293
def test_is_a():

0 commit comments

Comments
 (0)