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 | |
from py.test import raises |
from py.magic import patch, revert |
|
def test_patch_revert(): |
class a: |
pass |
raises(AttributeError, "patch(a, 'i', 42)") |
|
a.i = 42 |
patch(a, 'i', 23) |
assert a.i == 23 |
revert(a, 'i') |
assert a.i == 42 |
|
def test_double_patch(): |
class a: |
i = 42 |
assert patch(a, 'i', 2) == 42 |
assert patch(a, 'i', 3) == 2 |
assert a.i == 3 |
assert revert(a, 'i') == 3 |
assert a.i == 2 |
assert revert(a, 'i') == 2 |
assert a.i == 42 |
|
def test_valueerror(): |
class a: |
i = 2 |
pass |
raises(ValueError, "revert(a, 'i')") |
|
| |