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
135
136
137
138
139
140
|
from unittest import TestCase
from types import SimpleNamespace
import gc
import weakref
from pgzero.animation import animate
from pgzero import clock
class TestObject:
"""A simple object subclass"""
class SuggestionTest(TestCase):
def test_basic_animation(self):
"""Test that animation works"""
obj = SimpleNamespace()
obj.attribute = 0
animate(obj, attribute=2, duration=2)
self.assertEqual(obj.attribute, 0)
clock.tick(1)
self.assertEqual(obj.attribute, 1)
clock.tick(1)
self.assertEqual(obj.attribute, 2)
clock.tick(1)
self.assertEqual(obj.attribute, 2)
def test_some_tween(self):
"""Test that tweening does something"""
obj = SimpleNamespace()
obj.attribute = 0
animate(obj, attribute=2, tween='accelerate', duration=2)
self.assertEqual(obj.attribute, 0)
clock.tick(1)
assert 0 < obj.attribute < 1
clock.tick(1)
self.assertEqual(obj.attribute, 2)
clock.tick(1)
self.assertEqual(obj.attribute, 2)
def test_tuple_animation(self):
"""Test that you can animate a tuple"""
obj = SimpleNamespace()
obj.attribute = 0, 2
animate(obj, attribute=(2, 0), duration=2)
self.assertEqual(obj.attribute, (0, 2))
clock.tick(1)
self.assertEqual(obj.attribute, (1, 1))
clock.tick(1)
self.assertEqual(obj.attribute, (2, 0))
clock.tick(1)
self.assertEqual(obj.attribute, (2, 0))
def test_list_animation(self):
"""Test that you can animate a list"""
obj = SimpleNamespace()
obj.attribute = [0, 2]
animate(obj, attribute=[2, 0], duration=2)
self.assertEqual(obj.attribute, [0, 2])
clock.tick(1)
self.assertEqual(obj.attribute, [1, 1])
clock.tick(1)
self.assertEqual(obj.attribute, [2, 0])
clock.tick(1)
self.assertEqual(obj.attribute, [2, 0])
def test_on_finished(self):
"""Test the on_finished callback works"""
obj = SimpleNamespace()
obj.attribute = 0
endlist = ['in progress']
animate(obj, attribute=2, duration=2, on_finished=endlist.clear)
self.assertEqual(obj.attribute, 0)
self.assertEqual(endlist, ['in progress'])
clock.tick(1)
self.assertEqual(obj.attribute, 1)
self.assertEqual(endlist, ['in progress'])
clock.tick(1)
self.assertEqual(obj.attribute, 2)
clock.tick(1)
self.assertEqual(obj.attribute, 2)
self.assertEqual(endlist, [])
def test_cancel_old(self):
"""Test an old animation is cancelled when overwritten"""
obj = SimpleNamespace()
obj.attribute = 0
animate(obj, attribute=-4, duration=4)
animate(obj, attribute=2, duration=2)
self.assertEqual(obj.attribute, 0)
clock.tick(1)
self.assertEqual(obj.attribute, 1)
clock.tick(1)
self.assertEqual(obj.attribute, 2)
clock.tick(1)
self.assertEqual(obj.attribute, 2)
def test_cancel_some(self):
"""Test only the overwritten target of an old animation is cancelled"""
obj = SimpleNamespace()
obj.attr1 = 0
obj.attr2 = 0
animate(obj, attr1=-4, attr2=-4, duration=4)
animate(obj, attr1=2, duration=2)
self.assertEqual(obj.attr1, 0)
self.assertEqual(obj.attr2, 0)
clock.tick(1)
self.assertEqual(obj.attr1, 1)
self.assertEqual(obj.attr2, -1)
clock.tick(1)
self.assertEqual(obj.attr1, 2)
self.assertEqual(obj.attr2, -2)
clock.tick(1)
self.assertEqual(obj.attr1, 2)
self.assertEqual(obj.attr2, -3)
def test_no_linger(self):
"""Test that deleted animations are garbage-collected"""
anim1_alivelist = ['animation 1 alive']
anim2_alivelist = ['animation 2 alive']
obj_alivelist = ['object alive']
# cannot use SimpleNamespace because it doesn't support weakref
obj = TestObject()
obj.attribute = 0
anim1 = animate(obj, attribute=1, duration=5)
anim2 = animate(obj, attribute=2, duration=1)
weakref.finalize(anim1, anim1_alivelist.clear)
weakref.finalize(anim2, anim2_alivelist.clear)
weakref.finalize(obj, obj_alivelist.clear)
del anim1
del anim2
del obj
gc.collect()
self.assertEqual(anim1_alivelist, [])
clock.tick(3)
gc.collect()
self.assertEqual(anim2_alivelist, [])
self.assertEqual(obj_alivelist, [])
|
