[style] update python style to conform to PEP 8 (#3951)

With the exception of line length set to 119 vs. 79.

Add tests/ and tools/ to py-pretty-check.
This commit is contained in:
Jonathan Hui
2019-07-11 11:45:55 -04:00
committed by GitHub
parent a938ee2845
commit f924adcb60
341 changed files with 8702 additions and 4460 deletions
+80 -32
View File
@@ -45,7 +45,9 @@ class TestMacParser(unittest.TestCase):
frame = mac802154.MacFrame()
frame.parse(io.BytesIO(bytearray([0x12, 0x00, 0x12, 0x34, 0x56])))
self.assertEqual(mac802154.MacHeader.FrameType.ACK, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.ACK,
frame.header.frame_type)
self.assertEqual(True, frame.header.frame_pending)
self.assertEqual(False, frame.header.ack_request)
self.assertEqual(0, frame.header.frame_version)
@@ -58,7 +60,9 @@ class TestMacParser(unittest.TestCase):
frame.parse(io.BytesIO(bytearray([0x61, 0x88, 0x34, 0xce, 0xfa,
0xad, 0xde, 0xef, 0xbe, 0x12, 0x34, 0xfe, 0xdc])))
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.DATA,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(0, frame.header.frame_version)
@@ -81,18 +85,30 @@ class TestMacParser(unittest.TestCase):
0x12, 0x34,
0xfe, 0xdc])))
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.DATA,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(0, frame.header.frame_version)
self.assertEqual(0x56, frame.header.seq)
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
self.assertEqual(0xface, frame.header.dest_pan_id)
self.assertEqual(bytearray(reversed([0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef])),
frame.header.dest_address.mac_address)
self.assertEqual(
bytearray(
reversed(
[
0xde, 0xad, 0xbe, 0xef,
0xde, 0xad, 0xbe, 0xef])),
frame.header.dest_address.mac_address)
self.assertEqual(0xface, frame.header.src_pan_id)
self.assertEqual(bytearray(reversed([0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d])),
frame.header.src_address.mac_address)
self.assertEqual(
bytearray(
reversed(
[
0x0b, 0xad, 0xf0, 0x0d,
0xba, 0xd0, 0xd0, 0x0d])),
frame.header.src_address.mac_address)
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
@@ -106,7 +122,9 @@ class TestMacParser(unittest.TestCase):
0x12, 0x34,
0xfe, 0xdc])))
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.DATA,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(0, frame.header.frame_version)
@@ -115,8 +133,13 @@ class TestMacParser(unittest.TestCase):
self.assertEqual(0xface, frame.header.dest_pan_id)
self.assertEqual(0xdead, frame.header.dest_address.rloc)
self.assertEqual(0xface, frame.header.src_pan_id)
self.assertEqual(bytearray(reversed([0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d])),
frame.header.src_address.mac_address)
self.assertEqual(
bytearray(
reversed(
[
0x0b, 0xad, 0xf0, 0x0d,
0xba, 0xd0, 0xd0, 0x0d])),
frame.header.src_address.mac_address)
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
@@ -125,20 +148,28 @@ class TestMacParser(unittest.TestCase):
frame.parse(io.BytesIO(bytearray([0x61, 0x8c,
0x56,
0xce, 0xfa,
0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef,
0xde, 0xad, 0xbe, 0xef,
0xde, 0xad, 0xbe, 0xef,
0x0d, 0xf0,
0x12, 0x34,
0xfe, 0xdc])))
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.DATA,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(0, frame.header.frame_version)
self.assertEqual(0x56, frame.header.seq)
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
self.assertEqual(0xface, frame.header.dest_pan_id)
self.assertEqual(bytearray(reversed([0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef])),
frame.header.dest_address.mac_address)
self.assertEqual(
bytearray(
reversed(
[
0xde, 0xad, 0xbe, 0xef,
0xde, 0xad, 0xbe, 0xef])),
frame.header.dest_address.mac_address)
self.assertEqual(0xface, frame.header.src_pan_id)
self.assertEqual(0xf00d, frame.header.src_address.rloc)
@@ -154,7 +185,9 @@ class TestMacParser(unittest.TestCase):
0x04,
0xfe, 0xdc])))
self.assertEqual(mac802154.MacHeader.FrameType.COMMAND, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.COMMAND,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(0, frame.header.frame_version)
@@ -169,8 +202,8 @@ class TestMacParser(unittest.TestCase):
def test_should_decrypt_data_frame(self):
mac802154.DeviceDescriptors.add(0x2001, MacAddress(bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x07]),
MacAddressType.LONG))
mac802154.DeviceDescriptors.add(0x2001, MacAddress(bytearray(
[0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x07]), MacAddressType.LONG))
frame = mac802154.MacFrame()
frame.parse(io.BytesIO(bytearray([0x69, 0x98, 0x68, # FC, seq
@@ -182,11 +215,13 @@ class TestMacParser(unittest.TestCase):
0xc4, 0x6f, 0x7d, 0x6b, 0xb5, 0x4a, 0x87, 0x14,
0xae, 0xdd, 0x8e, 0xb7, 0x37, 0x62, 0x27, 0x48,
0xc9, 0x53, 0x0c, 0x44, 0x31, 0x59, 0x8b, 0xa2,
0x83, 0x59, 0xa1, 0x43,
0x74, 0xe0, 0x2a, 0xf6, # MIC (valid)
0x83, 0x59, 0xa1, 0x43, # MIC (valid)
0x74, 0xe0, 0x2a, 0xf6,
0x99, 0xfc]))) # FCS (valid)
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.DATA,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(1, frame.header.frame_version)
@@ -200,31 +235,44 @@ class TestMacParser(unittest.TestCase):
self.assertEqual(0, frame.header.aux_sec_header.frame_counter)
self.assertEqual(5, frame.header.aux_sec_header.security_level)
self.assertEqual(bytes(bytearray([0x7c, 0x77, 0x80, 0xf0, 0x4d, 0x4d, 0x4d, 0x4d,
0xe0, 0x04, 0x44, 0x02, 0x44, 0x66, 0x13, 0x5f,
0x22, 0x80, 0xb1, 0x61, 0x02, 0x61, 0x73, 0x11,
0x2a, 0xff, 0x01, 0x08, 0x16, 0x6e, 0x0a, 0x00,
0x00, 0x00, 0x00, 0x07])), frame.payload.data)
self.assertEqual(bytes(bytearray([0x7c, 0x77, 0x80, 0xf0,
0x4d, 0x4d, 0x4d, 0x4d,
0xe0, 0x04, 0x44, 0x02,
0x44, 0x66, 0x13, 0x5f,
0x22, 0x80, 0xb1, 0x61,
0x02, 0x61, 0x73, 0x11,
0x2a, 0xff, 0x01, 0x08,
0x16, 0x6e, 0x0a, 0x00,
0x00, 0x00, 0x00, 0x07])),
frame.payload.data)
def test_should_decrypt_command_frame(self):
frame = mac802154.MacFrame()
frame.parse(io.BytesIO(bytearray([0x6b, 0xdc, 0xce, 0xce, 0xfa, 0x02, 0x00, 0x00,
0x00, 0x00, 0x0a, 0x6e, 0x16, 0x03, 0x00, 0x00,
0x00, 0x00, 0x0a, 0x6e, 0x16, 0x0d, 0x00, 0x00,
0x00, 0x00, 0x01, 0x04, 0x2d, 0xbc, 0x12, 0xbe,
frame.parse(io.BytesIO(bytearray([0x6b, 0xdc, 0xce, 0xce,
0xfa, 0x02, 0x00, 0x00,
0x00, 0x00, 0x0a, 0x6e,
0x16, 0x03, 0x00, 0x00,
0x00, 0x00, 0x0a, 0x6e,
0x16, 0x0d, 0x00, 0x00,
0x00, 0x00, 0x01, 0x04,
0x2d, 0xbc, 0x12, 0xbe,
0x0a, 0x4f])))
self.assertEqual(mac802154.MacHeader.FrameType.COMMAND, frame.header.frame_type)
self.assertEqual(
mac802154.MacHeader.FrameType.COMMAND,
frame.header.frame_type)
self.assertEqual(False, frame.header.frame_pending)
self.assertEqual(True, frame.header.ack_request)
self.assertEqual(1, frame.header.frame_version)
self.assertEqual(206, frame.header.seq)
self.assertEqual(bytearray([0x0a, 0x4f]), frame.header.fcs)
self.assertEqual(0xface, frame.header.dest_pan_id)
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x02]),
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00,
0x00, 0x00, 0x00, 0x02]),
frame.header.dest_address.mac_address)
self.assertEqual(0xface, frame.header.src_pan_id)
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x03]),
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00,
0x00, 0x00, 0x00, 0x03]),
frame.header.src_address.mac_address)
self.assertEqual(0, frame.header.aux_sec_header.frame_counter)