Archives/wting_autojump
1
0
Fork 0
mirror of https://github.com/wting/autojump synced 2024-10-27 20:34:07 +00:00
Code Issues Releases Wiki Activity

modify iteration tests to explicitly work with generators

Browse Source
This commit is contained in:
William Ting 2013-12-31 19:09:51 -06:00
parent b3162c917b
commit 85824098c1
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/autojump_utils_test.py
View File

@ -59,22 +59,21 @@ class StringUnitTests(TestCase):
class IterationUnitTests(TestCase):
def test_first(self):
assert_equal(first((0, 1)), 0)
assert_equal(first(()), None)
assert_equal(first(xrange(5)), 0)
assert_equal(first([]), None)
def test_second(self):
assert_equal(second((0, 1)), 1)
assert_equal(second((0,)), None)
assert_equal(second(xrange(5)), 1)
assert_equal(second([]), None)
def test_last(self):
assert_equal(last((1, 2, 3)), 3)
assert_equal(last(()), None)
assert_equal(last(xrange(4)), 3)
assert_equal(last([]), None)
def test_take(self):
xs = [1, 2, 3]
assert_equal(list(take(1, xs)), [1])
assert_equal(list(take(2, xs)), [1, 2])
assert_equal(list(take(4, xs)), [1, 2, 3])
assert_equal(list(take(1, xrange(3))), [0])
assert_equal(list(take(2, xrange(3))), [0, 1])
assert_equal(list(take(4, xrange(3))), [0, 1, 2])
assert_equal(list(take(10, [])), [])