Bladeren bron

modify iteration tests to explicitly work with generators

pull/252/head
William Ting 10 jaren geleden
bovenliggende
commit
85824098c1
1 gewijzigde bestanden met toevoegingen van 9 en 10 verwijderingen
  1. +9
    -10
      tests/autojump_utils_test.py

+ 9
- 10
tests/autojump_utils_test.py Bestand weergeven

@ -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, [])), [])

Laden…
Annuleren
Opslaan