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
|
#!/usr/bin/python
import unittest
import lcmtest
class TestBools(unittest.TestCase):
def test_bool(self):
"""Encode a bools_t message, then verify that it decodes correctly.
Also check that the decoded fields are all of type bool.
"""
msg = lcmtest.bools_t()
msg.one_bool = True
msg.fixed_array = [False, True, False]
msg.num_a = 3
msg.num_b = 2
for a_index in xrange(msg.num_a):
inner_list = []
for b_index in xrange(msg.num_b):
inner_list.append(bool(b_index % 2))
msg.two_dim_array.append(inner_list)
msg.one_dim_array.append(bool((a_index + 1) % 2))
data = msg.encode()
decoded = lcmtest.bools_t.decode(data)
self.assertEqual(msg.one_bool, decoded.one_bool)
self.assertEqual(list(msg.fixed_array), list(decoded.fixed_array))
self.assertEqual(msg.num_a, decoded.num_a)
self.assertEqual(msg.num_b, decoded.num_b)
self.assertEqual(bool, type(decoded.one_bool))
self.assertTrue(all([type(elem) == bool
for elem in decoded.fixed_array]))
for sublist in decoded.two_dim_array:
self.assertTrue(all([type(elem) == bool for elem in sublist]))
self.assertTrue(all([type(elem) == bool
for elem in decoded.one_dim_array]))
for a_index in xrange(msg.num_a):
for b_index in xrange(msg.num_b):
self.assertEqual(msg.two_dim_array[a_index][b_index],
decoded.two_dim_array[a_index][b_index])
self.assertEqual(msg.one_dim_array[a_index],
decoded.one_dim_array[a_index])
if __name__ == '__main__':
unittest.main()
|
