make assign assertion more concise

Author: rerpha

tests/test_listen.py

@@ -1,5 +1,7 @@
 from unittest import mock
 
+from confluent_kafka import TopicPartition
+
 from saluki.listen import listen
 
 
@@ -14,8 +16,7 @@ def test_listen_with_partition_assigns_to_partition():
         mock.patch("saluki.listen.Consumer") as c,
     ):
         listen("somebroker", "sometopic", partition=expected_partition)
-        assert c.return_value.assign.call_args[0][0][0].topic == topic
-        assert c.return_value.assign.call_args[0][0][0].partition == expected_partition
+        c.return_value.assign.assert_called_with([TopicPartition(topic, expected_partition)])
 
 
 def test_keyboard_interrupt_causes_consumer_to_close():