aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/func/ut
diff options
context:
space:
mode:
authorrnefyodov <rnefyodov@yandex-team.ru>2022-02-10 16:47:17 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:47:17 +0300
commitc753751b693cf7c481c0292912e2b7536fa6d36a (patch)
tree9814fbd1c3effac9b8377c5d604b367b14e2db55 /library/python/func/ut
parentc22320e8c4f3d7be38c504706f137034e91d31e6 (diff)
downloadydb-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.py144
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__])