From 8c1ad64464b7128dca6fc073813d37af64327634 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Sat, 7 May 2022 00:29:58 +0200 Subject: [PATCH] refactoring; additional buffer test --- pyinfra/server/processor/processor.py | 2 +- test/unit_tests/server/__init__.py | 0 test/unit_tests/{ => server}/buffer_test.py | 7 +++++++ test/unit_tests/{ => server}/pipeline_test.py | 0 4 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 test/unit_tests/server/__init__.py rename test/unit_tests/{ => server}/buffer_test.py (74%) rename test/unit_tests/{ => server}/pipeline_test.py (100%) diff --git a/pyinfra/server/processor/processor.py b/pyinfra/server/processor/processor.py index 4321011..80cd220 100644 --- a/pyinfra/server/processor/processor.py +++ b/pyinfra/server/processor/processor.py @@ -1,5 +1,5 @@ from itertools import chain -from typing import Union, Any, Iterable, Callable +from typing import Union, Any from pyinfra.server.dispatcher.dispatcher import Nothing diff --git a/test/unit_tests/server/__init__.py b/test/unit_tests/server/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/test/unit_tests/buffer_test.py b/test/unit_tests/server/buffer_test.py similarity index 74% rename from test/unit_tests/buffer_test.py rename to test/unit_tests/server/buffer_test.py index ad929fc..adba909 100644 --- a/test/unit_tests/buffer_test.py +++ b/test/unit_tests/server/buffer_test.py @@ -17,3 +17,10 @@ def test_buffer(): reverse_buffer = bufferize(compose(reverse_buffer, list), buffer_size=3) ys = lmapcat(reverse_buffer, range(10)) assert ys == [2, 1, 0, 5, 4, 3, 8, 7, 6] + + def buffer_sum(xs): + return [sum(xs)] + + buffer_sum = bufferize(buffer_sum, buffer_size=3) + ys = lmapcat(buffer_sum, range(10)) + assert ys == [0, 0, 3, 0, 0, 12, 0, 0, 21, 0] diff --git a/test/unit_tests/pipeline_test.py b/test/unit_tests/server/pipeline_test.py similarity index 100% rename from test/unit_tests/pipeline_test.py rename to test/unit_tests/server/pipeline_test.py