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