diff options
author | rnefyodov <rnefyodov@yandex-team.ru> | 2022-02-10 16:47:17 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:47:17 +0300 |
commit | c753751b693cf7c481c0292912e2b7536fa6d36a (patch) | |
tree | 9814fbd1c3effac9b8377c5d604b367b14e2db55 /library/python/func/ut | |
parent | c22320e8c4f3d7be38c504706f137034e91d31e6 (diff) | |
download | ydb-c753751b693cf7c481c0292912e2b7536fa6d36a.tar.gz |
Restoring authorship annotation for <rnefyodov@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/python/func/ut')
-rw-r--r-- | library/python/func/ut/test_func.py | 144 |
1 files changed, 72 insertions, 72 deletions
diff --git a/library/python/func/ut/test_func.py b/library/python/func/ut/test_func.py index f3605bafcf..3c4fad1a07 100644 --- a/library/python/func/ut/test_func.py +++ b/library/python/func/ut/test_func.py @@ -1,87 +1,87 @@ -import pytest +import pytest import threading - + import library.python.func as func - - -def test_map0(): + + +def test_map0(): assert None is func.map0(lambda x: x + 1, None) - assert 3 == func.map0(lambda x: x + 1, 2) + assert 3 == func.map0(lambda x: x + 1, 2) assert None is func.map0(len, None) - assert 2 == func.map0(len, [1, 2]) - - -def test_single(): - assert 1 == func.single([1]) - with pytest.raises(Exception): - assert 1 == func.single([]) - with pytest.raises(Exception): - assert 1 == func.single([1, 2]) - - -def test_memoize(): - class Counter(object): - @staticmethod - def inc(): - Counter._qty = getattr(Counter, '_qty', 0) + 1 - return Counter._qty - + assert 2 == func.map0(len, [1, 2]) + + +def test_single(): + assert 1 == func.single([1]) + with pytest.raises(Exception): + assert 1 == func.single([]) + with pytest.raises(Exception): + assert 1 == func.single([1, 2]) + + +def test_memoize(): + class Counter(object): + @staticmethod + def inc(): + Counter._qty = getattr(Counter, '_qty', 0) + 1 + return Counter._qty + @func.memoize() - def t1(a): - return a, Counter.inc() - + def t1(a): + return a, Counter.inc() + @func.memoize() - def t2(a): - return a, Counter.inc() - + def t2(a): + return a, Counter.inc() + @func.memoize() - def t3(a): - return a, Counter.inc() - + def t3(a): + return a, Counter.inc() + @func.memoize() - def t4(a): - return a, Counter.inc() - + def t4(a): + return a, Counter.inc() + @func.memoize() - def t5(a, b, c): - return a + b + c, Counter.inc() - + def t5(a, b, c): + return a + b + c, Counter.inc() + @func.memoize() - def t6(): - return Counter.inc() - + def t6(): + return Counter.inc() + @func.memoize(limit=2) def t7(a, _b): return a, Counter.inc() - assert (1, 1) == t1(1) - assert (1, 1) == t1(1) - assert (2, 2) == t1(2) - assert (2, 2) == t1(2) - - assert (1, 3) == t2(1) - assert (1, 3) == t2(1) - assert (2, 4) == t2(2) - assert (2, 4) == t2(2) - - assert (1, 5) == t3(1) - assert (1, 5) == t3(1) - assert (2, 6) == t3(2) - assert (2, 6) == t3(2) - - assert (1, 7) == t4(1) - assert (1, 7) == t4(1) - assert (2, 8) == t4(2) - assert (2, 8) == t4(2) - - assert (6, 9) == t5(1, 2, 3) - assert (6, 9) == t5(1, 2, 3) - assert (7, 10) == t5(1, 2, 4) - assert (7, 10) == t5(1, 2, 4) - - assert 11 == t6() - assert 11 == t6() - + assert (1, 1) == t1(1) + assert (1, 1) == t1(1) + assert (2, 2) == t1(2) + assert (2, 2) == t1(2) + + assert (1, 3) == t2(1) + assert (1, 3) == t2(1) + assert (2, 4) == t2(2) + assert (2, 4) == t2(2) + + assert (1, 5) == t3(1) + assert (1, 5) == t3(1) + assert (2, 6) == t3(2) + assert (2, 6) == t3(2) + + assert (1, 7) == t4(1) + assert (1, 7) == t4(1) + assert (2, 8) == t4(2) + assert (2, 8) == t4(2) + + assert (6, 9) == t5(1, 2, 3) + assert (6, 9) == t5(1, 2, 3) + assert (7, 10) == t5(1, 2, 4) + assert (7, 10) == t5(1, 2, 4) + + assert 11 == t6() + assert 11 == t6() + assert (1, 12) == t7(1, None) assert (2, 13) == t7(2, None) assert (1, 12) == t7(1, None) @@ -158,5 +158,5 @@ def test_memoize_thread_local(): th.join() -if __name__ == '__main__': - pytest.main([__file__]) +if __name__ == '__main__': + pytest.main([__file__]) |