Exemple #1
0
def test_oddity():
    """Test oddities of take."""
    data = iter("ABCDE")
    # specifying None as the length consumes the whole iterator
    res = take(None, data)
    assert res == ["A", "B", "C", "D", "E"]
    assert list(data) == []

    data = ()
    # specifying None as the length consumes the whole iterator
    res = take(None, data)
    assert res == []
Exemple #2
0
def test_normal():
    """Test take."""
    # take less than data
    data = iter("ABCDE")
    res = take(3, data)
    assert res == ["A", "B", "C"]
    assert list(data) == ["D", "E"]

    # take exact count of data
    data = iter("ABCDE")
    res = take(5, data)
    assert res == ["A", "B", "C", "D", "E"]
    assert list(data) == []

    # take more than data
    data = iter("ABCDE")
    res = take(7, data)
    assert res == ["A", "B", "C", "D", "E"]
    assert list(data) == []

    # take nothing
    data = iter("ABCDE")
    res = take(0, data)
    assert res == []
    assert list(data) == ["A", "B", "C", "D", "E"]

    # empty but request some
    data = ()
    res = take(3, data)
    assert res == []

    # empty and request none
    data = ()
    res = take(0, data)
    assert res == []