async/async/test/test_example.py at master · gitpython-developers/async

This repository was archived by the owner on Apr 14, 2024. It is now read-only.

45 lines (33 loc) · 1.42 KB

# This module is part of async and is released under

# the New BSD License: http://www.opensource.org/licenses/bsd-license.php

"""Module containing examples from the documentaiton"""

from .lib import TestBase

from async.pool import ThreadPool

from async.task import (

IteratorThreadTask,

ChannelThreadTask

)

from async.thread import terminate_threads

class TestExamples(TestBase):

@terminate_threads

def test_usage(self):

p = ThreadPool()

# default size is 0, synchronous mode

assert p.size() == 0

# A task performing processing on items from an iterator

t = IteratorThreadTask(iter(list(range(10))), "power", lambda i: i*i)

reader = p.add_task(t)

# read all items - they where procesed by worker 1

items = reader.read()

assert len(items) == 10 and items[0] == 0 and items[-1] == 81

# chaining

t = IteratorThreadTask(iter(list(range(10))), "power", lambda i: i*i)

reader = p.add_task(t)

# chain both by linking their readers

tmult = ChannelThreadTask(reader, "mult", lambda i: i*2)

result_reader = p.add_task(tmult)

# read all

items = result_reader.read()

assert len(items) == 10 and items[0] == 0 and items[-1] == 162

Navigation Menu