1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
#!/usr/bin/env vpython3
# Copyright 2017 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Unit tests for decorators.py."""
import unittest
import decorators
class NoRaiseExceptionDecoratorTest(unittest.TestCase):
def testFunctionDoesNotRaiseException(self):
"""Tests that the |NoRaiseException| decorator catches exception."""
@decorators.NoRaiseException()
def raiseException():
raise Exception()
try:
raiseException()
except Exception: # pylint: disable=broad-except
self.fail('Exception was not caught by |NoRaiseException| decorator')
def testFunctionReturnsCorrectValues(self):
"""Tests that the |NoRaiseException| decorator returns correct values."""
@decorators.NoRaiseException(default_return_value=111)
def raiseException():
raise Exception()
@decorators.NoRaiseException(default_return_value=111)
def doesNotRaiseException():
return 999
self.assertEqual(raiseException(), 111)
self.assertEqual(doesNotRaiseException(), 999)
def testFunctionReturnsCorrectValuesWithExceptionType(self):
"""Tests that the |NoRaiseException| decorator returns correct values."""
@decorators.NoRaiseException(default_return_value=111,
exception_type=OSError)
def raiseException():
raise OSError('test')
@decorators.NoRaiseException(default_return_value=111,
exception_type=OSError)
def doesNotRaiseException():
return 999
@decorators.NoRaiseException(default_return_value=111,
exception_type=OSError)
def raiseOtherException():
raise ValueError('test')
@decorators.NoRaiseException(default_return_value=111)
def raiseDefaultException():
raise Exception('test')
self.assertEqual(raiseException(), 111)
self.assertEqual(doesNotRaiseException(), 999)
self.assertEqual(raiseDefaultException(), 111)
with self.assertRaises(ValueError):
raiseOtherException()
class MemoizeDecoratorTest(unittest.TestCase):
def testFunctionExceptionNotMemoized(self):
"""Tests that |Memoize| decorator does not cache exception results."""
class ExceptionType1(Exception):
pass
class ExceptionType2(Exception):
pass
@decorators.Memoize
def raiseExceptions():
if raiseExceptions.count == 0:
raiseExceptions.count += 1
raise ExceptionType1()
if raiseExceptions.count == 1:
raise ExceptionType2()
raiseExceptions.count = 0
with self.assertRaises(ExceptionType1):
raiseExceptions()
with self.assertRaises(ExceptionType2):
raiseExceptions()
def testFunctionResultMemoized(self):
"""Tests that |Memoize| decorator caches results."""
@decorators.Memoize
def memoized():
memoized.count += 1
return memoized.count
memoized.count = 0
def notMemoized():
notMemoized.count += 1
return notMemoized.count
notMemoized.count = 0
self.assertEqual(memoized(), 1)
self.assertEqual(memoized(), 1)
self.assertEqual(memoized(), 1)
self.assertEqual(notMemoized(), 1)
self.assertEqual(notMemoized(), 2)
self.assertEqual(notMemoized(), 3)
def testFunctionMemoizedBasedOnArgs(self):
"""Tests that |Memoize| caches results based on args and kwargs."""
@decorators.Memoize
def returnValueBasedOnArgsKwargs(a, k=0):
return a + k
self.assertEqual(returnValueBasedOnArgsKwargs(1, 1), 2)
self.assertEqual(returnValueBasedOnArgsKwargs(1, 2), 3)
self.assertEqual(returnValueBasedOnArgsKwargs(2, 1), 3)
self.assertEqual(returnValueBasedOnArgsKwargs(3, 3), 6)
if __name__ == '__main__':
unittest.main(verbosity=2)
|
