reaplce some more assertEqual

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