Ejemplo n.º 1
0
def test_take():
    assert list(take(1, xrange(3))) == [0]
    assert list(take(2, xrange(3))) == [0, 1]
    assert list(take(4, xrange(3))) == [0, 1, 2]
    assert list(take(10, [])) == []
Ejemplo n.º 2
0
 def test_take(self):
     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, [])), [])
Ejemplo n.º 3
0
def test_take():
    assert list(take(1, xrange(3))) == [0]
    assert list(take(2, xrange(3))) == [0, 1]
    assert list(take(4, xrange(3))) == [0, 1, 2]
    assert list(take(10, [])) == []