Skip to content

Commit 8bddf59

Browse files
+ Adding processor test cases
1 parent 164d54e commit 8bddf59

File tree

1 file changed

+46
-1
lines changed

1 file changed

+46
-1
lines changed

tests/test_decorator.py

Lines changed: 46 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import time
2-
from src.thread import threaded
2+
from src.thread import threaded, processor
33

44

55
# >>>>>>>>>> Dummy Functions <<<<<<<<<< #
@@ -43,3 +43,48 @@ def _run(*args):
4343

4444
x = _run(8, 9)
4545
assert x.get_return_value() == (1, 2, 3, 8, 9)
46+
47+
48+
49+
50+
def test_processorCreationNoParam():
51+
@processor
52+
def _run(args):
53+
return _dummy_target_raiseToPower(*args)
54+
55+
x = _run([[2, 2]])
56+
assert x.get_return_values() == [4]
57+
58+
def test_processorCreationEmptyParam():
59+
@processor()
60+
def _run(args):
61+
return _dummy_target_raiseToPower(*args)
62+
63+
x = _run([[2, 2]])
64+
assert x.get_return_values() == [4]
65+
66+
def test_processorCreationWithParam():
67+
@processor(daemon = True)
68+
def _run(args):
69+
return _dummy_target_raiseToPower(*args)
70+
71+
x = _run([[2, 2]])
72+
assert len(x._threads) == 1
73+
assert x._threads[0].thread.daemon
74+
assert x.get_return_values() == [4]
75+
76+
def test_processorArgJoin():
77+
@processor(daemon = True, args = (1, 2, 3))
78+
def _run(data, *args):
79+
return [*args, *data]
80+
81+
x = _run([[8, 9]])
82+
assert x.get_return_values() == [[1, 2, 3, 8, 9]]
83+
84+
def test_processorMultiArgJoin():
85+
@processor(daemon = True, args = (1, 2, 3))
86+
def _run(data, *args):
87+
return [*args, *data]
88+
89+
x = _run([[8, 9], [10, 11]])
90+
assert x.get_return_values() == [[1, 2, 3, 8, 9], [1, 2, 3, 10, 11]]

0 commit comments

Comments
 (0)