def test_stream_accumulate():
    ''' test the accumulate function and what it expects as input.
    '''
    # define an acc
    def myacc(prevstate, newstate):
        ''' Accumulate on a state and return a next state.

            Note that accumulator could have returned a newstate, nextout pair.
            Howevever, in that case, an initializer needs to be defined.  This
            may be unnecessary overhead.
        '''
        nextstate = newstate + prevstate
        return nextstate

    s = Stream()
    sacc = s.accumulate(myacc)

    # check the start keyword
    sacc2 = s.accumulate(myacc, start=1)
    L = sacc.sink_to_list()

    L2 = sacc2.sink_to_list()

    s.emit(1)
    s.emit(1)
    s.emit(4)

    # print(L)

    # should emit on first
    assert L == [1, 2, 6]

    # should emit on first
    assert L2 == [2, 3, 7]
示例#2
0
def test_stream_accumulate():
    ''' This tests that the dispatching on the streamdoc's accumulate routine
    is working properly.'''
    def myacc(prevstate, newstate):
        return prevstate + newstate

    s = Stream()
    sout = s.accumulate(psda(myacc))

    L = sout.sink_to_list()

    sout.emit(StreamDoc(args=[1]))
    sout.emit(StreamDoc(args=[2]))
    sout.emit(StreamDoc(args=[3]))

    print(L)
示例#3
0
def test_accumulate_errors_raises():
    a = Stream()
    b = a.accumulate(lambda x, y: x / y)
    with pytest.raises(ZeroDivisionError):
        a.emit(1)
        a.emit(0)
示例#4
0
def example2():
    source = Stream()
    source.accumulate(add).sink(print)
    source.emit(1)
    source.emit(2)
    source.emit(3)
示例#5
0
def test_accumulate_errors_raises():
    a = Stream()
    b = a.accumulate(lambda x, y: x / y, with_state=True)  # noqa: F841
    with pytest.raises(ZeroDivisionError):
        a.emit(1)
        a.emit(0)