/
tests.py
74 lines (59 loc) · 2.69 KB
/
tests.py
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
import unittest
from GallopTracker import GallopTracker
class GallopTrackerTestCase(unittest.TestCase):
def setUp(self):
self.gallop_tracker = GallopTracker()
def tearDown(self):
self.gallop_tracker = None
def test_not_galloping_when_empty(self):
self.assertEqual(self.gallop_tracker.isGalloping(), False,
'should return false when empty')
def test_not_galloping_when_one(self):
self.gallop_tracker.addNoha(320, 320)
self.assertEqual(self.gallop_tracker.isGalloping(), False,
'should return false when one')
def test_galloping(self):
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 300)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 300)
self.gallop_tracker.addNoha(320, 340)
self.assertEqual(self.gallop_tracker.isGalloping(), True,'should Gallop')
def test_galloping_under_thresh(self):
self.gallop_tracker.addNoha(320, 325)
self.gallop_tracker.addNoha(320, 315)
self.gallop_tracker.addNoha(320, 325)
self.gallop_tracker.addNoha(320, 315)
self.gallop_tracker.addNoha(320, 325)
self.assertEqual(self.gallop_tracker.isGalloping(), False, 'should not Gallop')
def test_ring_buffer(self):
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 300)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 300)
self.gallop_tracker.addNoha(320, 340)
for i in range(100):
self.gallop_tracker.addNoha(320, 340)
self.assertEqual(self.gallop_tracker.isGalloping(), False, 'should not Gallop')
def test_galloping_lot_of_data(self):
# Same
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
#Switchy
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 300)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 300)
self.gallop_tracker.addNoha(320, 340)
# Same
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.gallop_tracker.addNoha(320, 340)
self.assertEqual(self.gallop_tracker.isGalloping(), True, 'should Gallop')
suite = unittest.TestLoader().loadTestsFromTestCase(GallopTrackerTestCase)
unittest.TextTestRunner(verbosity=2).run(suite)