diff --git a/tests/data/unittest.parquet b/tests/data/unittest.parquet new file mode 100644 index 0000000000..508abbb337 Binary files /dev/null and b/tests/data/unittest.parquet differ diff --git a/tests/execution_test.py b/tests/execution_test.py index f02d7d6447..14af86f5a1 100644 --- a/tests/execution_test.py +++ b/tests/execution_test.py @@ -373,6 +373,27 @@ async def means(): + +def test_mixed_pre_filtered() -> None: + df = vaex.from_arrays(x=np.arange(10)) + df['y'] = df.x**2 + with vaex.cache.off(): + @vaex.delayed + def add(a, b): + return a + b + + df2 = df[df.x > 2] + p1 = df2.sum(df.x, delay=True) + p2 = df2.nop(df.y, delay=True) + passes = df.executor.passes + tasks = df.executor.tasks.copy() + assert len(df.executor.tasks) == 2 + df.execute() + assert df.executor.passes == passes + 1 + # make sure we don't leave exceptions + p1.get() + p2.get() + # def test_add_and_cancel_tasks(df_executor): # df = df_executor diff --git a/tests/extract_test.py b/tests/extract_test.py index c4fb2290dc..c8b4b8a65a 100644 --- a/tests/extract_test.py +++ b/tests/extract_test.py @@ -1,3 +1,4 @@ +import threading from concurrent.futures import ThreadPoolExecutor from threading import Barrier import time @@ -33,3 +34,4 @@ def run(_ignore): dff.extract() pool = ThreadPoolExecutor(max_workers=100) _values = list(pool.map(run, range(100))) +