From f439ccd1b16f21b232f2f6df84e7365f3a0856ee Mon Sep 17 00:00:00 2001 From: Florian Ludwig Date: Sat, 6 Mar 2021 18:59:19 +0100 Subject: [PATCH] reaplce some more assertEqual --- tests/mqtt/test_publish.py | 34 +++++++++++++++++----------------- 1 file changed, 17 insertions(+), 17 deletions(-) diff --git a/tests/mqtt/test_publish.py b/tests/mqtt/test_publish.py index 06c2913..4d1a33d 100644 --- a/tests/mqtt/test_publish.py +++ b/tests/mqtt/test_publish.py @@ -20,8 +20,8 @@ class PublishPacketTest(unittest.TestCase): self.assertEqual(message.variable_header.topic_name, 'topic') self.assertEqual(message.variable_header.packet_id, None) self.assertFalse((message.fixed_header.flags >> 1) & 0x03) - self.assertTrue(message.fixed_header.flags & 0x01) - self.assertTrue(message.payload.data, b'0123456789') + assert message.fixed_header.flags & 0x01 + assert message.payload.data, b'0123456789' def test_from_stream_qos_2(self): data = b'\x37\x13\x00\x05topic\x00\x0a0123456789' @@ -29,9 +29,9 @@ class PublishPacketTest(unittest.TestCase): message = self.loop.run_until_complete(PublishPacket.from_stream(stream)) self.assertEqual(message.variable_header.topic_name, 'topic') self.assertEqual(message.variable_header.packet_id, 10) - self.assertTrue((message.fixed_header.flags >> 1) & 0x03) - self.assertTrue(message.fixed_header.flags & 0x01) - self.assertTrue(message.payload.data, b'0123456789') + assert (message.fixed_header.flags >> 1) & 0x03 + assert message.fixed_header.flags & 0x01 + assert message.payload.data, b'0123456789' def test_to_stream_no_packet_id(self): variable_header = PublishVariableHeader('topic', None) @@ -68,19 +68,19 @@ class PublishPacketTest(unittest.TestCase): packet = PublishPacket.build('/topic', b'data', 1, True, QOS_0, False) self.assertEqual(packet.packet_id, 1) - self.assertTrue(packet.dup_flag) + assert packet.dup_flag self.assertEqual(packet.qos, QOS_0) self.assertFalse(packet.retain_flag) packet = PublishPacket.build('/topic', b'data', 1, True, QOS_1, False) self.assertEqual(packet.packet_id, 1) - self.assertTrue(packet.dup_flag) + assert packet.dup_flag self.assertEqual(packet.qos, QOS_1) self.assertFalse(packet.retain_flag) packet = PublishPacket.build('/topic', b'data', 1, True, QOS_2, False) self.assertEqual(packet.packet_id, 1) - self.assertTrue(packet.dup_flag) + assert packet.dup_flag self.assertEqual(packet.qos, QOS_2) self.assertFalse(packet.retain_flag) @@ -88,34 +88,34 @@ class PublishPacketTest(unittest.TestCase): self.assertEqual(packet.packet_id, 1) self.assertFalse(packet.dup_flag) self.assertEqual(packet.qos, QOS_0) - self.assertTrue(packet.retain_flag) + assert packet.retain_flag packet = PublishPacket.build('/topic', b'data', 1, False, QOS_1, True) self.assertEqual(packet.packet_id, 1) self.assertFalse(packet.dup_flag) self.assertEqual(packet.qos, QOS_1) - self.assertTrue(packet.retain_flag) + assert packet.retain_flag packet = PublishPacket.build('/topic', b'data', 1, False, QOS_2, True) self.assertEqual(packet.packet_id, 1) self.assertFalse(packet.dup_flag) self.assertEqual(packet.qos, QOS_2) - self.assertTrue(packet.retain_flag) + assert packet.retain_flag packet = PublishPacket.build('/topic', b'data', 1, True, QOS_0, True) self.assertEqual(packet.packet_id, 1) - self.assertTrue(packet.dup_flag) + assert packet.dup_flag self.assertEqual(packet.qos, QOS_0) - self.assertTrue(packet.retain_flag) + assert packet.retain_flag packet = PublishPacket.build('/topic', b'data', 1, True, QOS_1, True) self.assertEqual(packet.packet_id, 1) - self.assertTrue(packet.dup_flag) + assert packet.dup_flag self.assertEqual(packet.qos, QOS_1) - self.assertTrue(packet.retain_flag) + assert packet.retain_flag packet = PublishPacket.build('/topic', b'data', 1, True, QOS_2, True) self.assertEqual(packet.packet_id, 1) - self.assertTrue(packet.dup_flag) + assert packet.dup_flag self.assertEqual(packet.qos, QOS_2) - self.assertTrue(packet.retain_flag) + assert packet.retain_flag \ No newline at end of file