def test_set_test_mode(self): spi = SPI(zdok=0, test=True) self.assertEqual(spi.regs, []) spi.set_test_mode() expRegs = [("0x85", "0x0"), ("0x81", "0x13c8")] self.assertEqual(expRegs, spi.get_hex_regs()) # writes: [(4, '\x00\x00\x85\x01'), (4, '\x13\xc8\x81\x01')] writes = spi.get_int_roach_writes() # [unpack(">I", data) for offset, data in spi.writes] expWrites = [(34049,), (331907329,)] self.assertEqual(expWrites, writes)
def test_set_control(self): spi = SPI(zdok=0, test=True) self.assertEqual(spi.regs, []) spi.set_control() self.assertEqual(len(spi.regs), 1) # regs = [(hex(int(x)), hex(int(y))) for x, y in spi.regs] exp = [("0x81", "0x3c8")] self.assertEqual(exp, spi.get_hex_regs()) # writes: [(4, '\x03\xc8\x81\x01')] writes = [unpack(">I", data) for offset, data in spi.writes] expWrites = [(63471873,)] # self.assertEqual(expWrites, writes)