class TestCodeBugGet(unittest.TestCase): def setUp(self): self.codebug = CodeBugRaw(serial.Serial('/dev/ttyACM0')) # self.codebug = CodeBugRaw(serial.Serial('/dev/pts/31')) self.num_channels = 5 def test_set_get(self): for i in range(self.num_channels): self.codebug.set(i, (i*i) % 0x1F) for i in range(self.num_channels): self.assertEqual(self.codebug.get(i), (i*i) % 0x1F)
def setUp(self): self.codebug = CodeBugRaw(serial.Serial('/dev/ttyACM0')) # self.codebug = CodeBugRaw(serial.Serial('/dev/pts/4')) self.num_channels = 5
class TestCodeBugRawObject(unittest.TestCase): def setUp(self): self.codebug = CodeBugRaw(serial.Serial('/dev/ttyACM0')) # self.codebug = CodeBugRaw(serial.Serial('/dev/pts/4')) self.num_channels = 5 # @unittest.skip def test_channels_get_set(self): """ >>> codebug.set(1, 2) >>> codebug.get(1) 2 """ for i in range(self.num_channels): self.codebug.set(i, (i*i) % 0x1F) for i in range(self.num_channels): self.assertEqual(self.codebug.get(i), (i*i) % 0x1F) # with or_mask for i in range(self.num_channels): self.codebug.set(i, 0x10, or_mask=True) for i in range(self.num_channels): self.assertEqual(self.codebug.get(i), (0x10 | (i*i)) % 0x1F) # @unittest.skip def test_channel_get_set_bulk(self): """ >>> codebug.set_bulk(0, [1, 2, 3]) >>> codebug.channels[1].value 2 """ v = list(range(self.num_channels)) self.codebug.set_bulk(0, v) self.assertEqual(tuple(self.codebug.get_bulk(0, len(v))), tuple(v)) v[2] = 0b11011 v[3] = 0b01001 self.codebug.set_bulk(2, v[2:4]) self.assertEqual(tuple(self.codebug.get_bulk(0, self.num_channels)), tuple(v)) v = list(range(self.num_channels)) self.codebug.set_bulk(0, v) self.assertEqual(tuple(self.codebug.get_bulk(0, len(v))), tuple(v)) v[2] |= 0b10001 v[3] |= 0b01001 self.codebug.set_bulk(2, v[2:4], or_mask=True) self.assertEqual(tuple(self.codebug.get_bulk(0, self.num_channels)), tuple(v)) def test_channel_get_set_bulk(self): x = self.codebug.get(5) a = (x >> 4) & 0x1 b = (x >> 5) & 0x1 print("A is {}, B is {}".format(a, b))