forked from davidszotten/prettyqr
/
tests.py
104 lines (75 loc) · 2.81 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
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
import unittest
from prettyqr.blobgrid import BlobGrid
from prettyqr.utils import pointwise
LEFT = 0
DOWN = 1
RIGHT = 2
UP = 3
class TestPointwise(unittest.TestCase):
def test_add(self):
self.assertEqual(pointwise('add', [1, 1], [2, 2]), [3, 3])
def test_tuple(self):
self.assertEqual(pointwise('add', [1, 1], (2, 2)), (3, 3))
def test_singleton(self):
self.assertEqual(pointwise('add', [1, 1], 1), [2, 2])
def test_empty(self):
self.assertEqual(pointwise('add', [], []), [])
def test_missing_operator(self):
self.assertRaises(AttributeError, pointwise, 'noop', [], [])
class TestBlobGrid(unittest.TestCase):
def setUp(self):
test_data = [
[1, 1, 0],
[1, 1, 0],
[0, 0, 1]]
class BlobGridTest(BlobGrid):
def get_value(self, x, y):
size = len(test_data)
if not (0 <= x < size) or not (0 <= y < size):
return 0
return test_data[y][x]
self.blob_grid = BlobGridTest(3)
self.drawing = self.blob_grid.draw_blobs()
def test_no_subclass(self):
blob_grid = BlobGrid(0)
self.assertRaises(NotImplementedError, blob_grid.draw_blobs)
def test_can_move(self):
# counter clockwise
self.assertTrue(self.blob_grid.can_move((0, 0), DOWN))
def test_cant_move(self):
# clockwise
self.assertFalse(self.blob_grid.can_move((0, 0), RIGHT))
# outside
self.assertFalse(self.blob_grid.can_move((0, 0), UP))
self.assertFalse(self.blob_grid.can_move((0, 0), LEFT))
def test_find_direction1(self):
self.assertEquals(self.blob_grid.find_direction((0, 0)), DOWN)
def test_find_direction2(self):
self.assertEquals(
self.blob_grid.find_direction((2, 2)), DOWN)
def test_find_direction3(self):
self.assertEquals(
self.blob_grid.find_direction((2, 2), LEFT), DOWN)
def test_find_direction4(self):
self.assertEquals(
self.blob_grid.find_direction((2, 2), RIGHT), UP)
def test_move(self):
self.assertEqual(
self.blob_grid.move((0, 0), RIGHT), (1, 0))
def test_draw(self):
self.assertEqual(self.blob_grid.draw(LEFT, DOWN),
'c -0.276142374915,0.0 -0.5,0.276142374915 -0.5,0.5 ')
def test_draw_blobs_lines(self):
self.assertEquals(
self.drawing.count('\n'), 2)
def test_draw_blobs_corners(self):
self.assertEqual(
self.drawing.count('c '), 8)
def test_draw_blobs_lines(self):
self.assertEqual(
self.drawing.count('l '), 4)
def test_draw_blobs_colour_singles(self):
self.assertEqual(
self.drawing.count('colour'), 1)
if __name__ == '__main__':
unittest.main()