run black w/ extended line length

tests/pubsub/conftest.py

@@ -16,9 +16,7 @@ def num_hosts():
 @pytest.fixture
 async def hosts(num_hosts):
     _hosts = HostFactory.create_batch(num_hosts)
-    await asyncio.gather(
-        *[_host.get_network().listen(LISTEN_MADDR) for _host in _hosts]
-    )
+    await asyncio.gather(*[_host.get_network().listen(LISTEN_MADDR) for _host in _hosts])
     yield _hosts
     # Clean up
     listeners = []

tests/pubsub/floodsub_integration_test_settings.py

@@ -102,10 +102,7 @@ FLOODSUB_PROTOCOL_TEST_CASES = [
             "3": ["1", "2", "4"],
             "4": ["1", "2", "3"],
         },
-        "topic_map": {
-            "astrophysics": ["1", "2", "3", "4"],
-            "school": ["1", "2", "3", "4"],
-        },
+        "topic_map": {"astrophysics": ["1", "2", "3", "4"], "school": ["1", "2", "3", "4"]},
         "messages": [
             {"topics": ["astrophysics"], "data": b"e=mc^2", "node_id": "1"},
             {"topics": ["school"], "data": b"foobar", "node_id": "2"},
@@ -137,8 +134,7 @@ FLOODSUB_PROTOCOL_TEST_CASES = [
 ]
 
 floodsub_protocol_pytest_params = [
-    pytest.param(test_case, id=test_case["name"])
-    for test_case in FLOODSUB_PROTOCOL_TEST_CASES
+    pytest.param(test_case, id=test_case["name"]) for test_case in FLOODSUB_PROTOCOL_TEST_CASES
 ]
 
 
@@ -195,9 +191,7 @@ async def perform_test_from_obj(obj, router_factory):
             # Create neighbor if neighbor does not yet exist
             if neighbor_id not in node_map:
                 await add_node(neighbor_id)
-            tasks_connect.append(
-                connect(node_map[start_node_id], node_map[neighbor_id])
-            )
+            tasks_connect.append(connect(node_map[start_node_id], node_map[neighbor_id]))
     # Connect nodes and wait at least for 2 seconds
     await asyncio.gather(*tasks_connect, asyncio.sleep(2))
 

tests/pubsub/test_dummyaccount_demo.py

@@ -35,9 +35,7 @@ async def perform_test(num_nodes, adjacency_map, action_func, assertion_func):
     for source_num in adjacency_map:
         target_nums = adjacency_map[source_num]
         for target_num in target_nums:
-            await connect(
-                dummy_nodes[source_num].libp2p_node, dummy_nodes[target_num].libp2p_node
-            )
+            await connect(dummy_nodes[source_num].libp2p_node, dummy_nodes[target_num].libp2p_node)
 
     # Allow time for network creation to take place
     await asyncio.sleep(0.25)

tests/pubsub/test_gossipsub.py

@@ -10,8 +10,7 @@ from .utils import dense_connect, one_to_all_connect
 
 
 @pytest.mark.parametrize(
-    "num_hosts, gossipsub_params",
-    ((4, GossipsubParams(degree=4, degree_low=3, degree_high=5)),),
+    "num_hosts, gossipsub_params", ((4, GossipsubParams(degree=4, degree_low=3, degree_high=5)),)
 )
 @pytest.mark.asyncio
 async def test_join(num_hosts, hosts, gossipsubs, pubsubs_gsub):
@@ -343,11 +342,7 @@ async def test_fanout_maintenance(hosts, pubsubs_gsub):
         (
             2,
             GossipsubParams(
-                degree=1,
-                degree_low=0,
-                degree_high=2,
-                gossip_window=50,
-                gossip_history=100,
+                degree=1, degree_low=0, degree_high=2, gossip_window=50, gossip_history=100
             ),
         ),
     ),

tests/pubsub/test_gossipsub_backward_compatibility.py

@@ -20,7 +20,5 @@ async def test_gossipsub_initialize_with_floodsub_protocol():
 async def test_gossipsub_run_with_floodsub_tests(test_case_obj):
     await perform_test_from_obj(
         test_case_obj,
-        functools.partial(
-            GossipsubFactory, degree=3, degree_low=2, degree_high=4, time_to_live=30
-        ),
+        functools.partial(GossipsubFactory, degree=3, degree_low=2, degree_high=4, time_to_live=30),
     )

tests/pubsub/test_pubsub.py

@@ -128,9 +128,7 @@ async def test_continuously_read_stream(pubsubs_fsub, monkeypatch):
         event_handle_rpc.set()
 
     monkeypatch.setattr(pubsubs_fsub[0], "push_msg", mock_push_msg)
-    monkeypatch.setattr(
-        pubsubs_fsub[0], "handle_subscription", mock_handle_subscription
-    )
+    monkeypatch.setattr(pubsubs_fsub[0], "handle_subscription", mock_handle_subscription)
     monkeypatch.setattr(pubsubs_fsub[0].router, "handle_rpc", mock_handle_rpc)
 
     async def wait_for_event_occurring(event):
@@ -149,9 +147,7 @@ async def test_continuously_read_stream(pubsubs_fsub, monkeypatch):
     task = asyncio.ensure_future(pubsubs_fsub[0].continuously_read_stream(stream))
 
     # Test: `push_msg` is called when publishing to a subscribed topic.
-    publish_subscribed_topic = rpc_pb2.RPC(
-        publish=[rpc_pb2.Message(topicIDs=[TESTING_TOPIC])]
-    )
+    publish_subscribed_topic = rpc_pb2.RPC(publish=[rpc_pb2.Message(topicIDs=[TESTING_TOPIC])])
     await stream.write(publish_subscribed_topic.SerializeToString())
     await wait_for_event_occurring(event_push_msg)
     # Make sure the other events are not emitted.
@@ -204,10 +200,7 @@ def test_handle_subscription(pubsubs_fsub):
     peer_ids = [ID(b"\x12\x20" + i.to_bytes(32, "big")) for i in range(2)]
     # Test: One peer is subscribed
     pubsubs_fsub[0].handle_subscription(peer_ids[0], sub_msg_0)
-    assert (
-        len(pubsubs_fsub[0].peer_topics) == 1
-        and TESTING_TOPIC in pubsubs_fsub[0].peer_topics
-    )
+    assert len(pubsubs_fsub[0].peer_topics) == 1 and TESTING_TOPIC in pubsubs_fsub[0].peer_topics
     assert len(pubsubs_fsub[0].peer_topics[TESTING_TOPIC]) == 1
     assert peer_ids[0] in pubsubs_fsub[0].peer_topics[TESTING_TOPIC]
     # Test: Another peer is subscribed
@@ -233,10 +226,7 @@ def test_handle_subscription(pubsubs_fsub):
 async def test_handle_talk(pubsubs_fsub):
     sub = await pubsubs_fsub[0].subscribe(TESTING_TOPIC)
     msg_0 = make_pubsub_msg(
-        origin_id=pubsubs_fsub[0].my_id,
-        topic_ids=[TESTING_TOPIC],
-        data=b"1234",
-        seqno=b"\x00" * 8,
+        origin_id=pubsubs_fsub[0].my_id, topic_ids=[TESTING_TOPIC], data=b"1234", seqno=b"\x00" * 8
     )
     await pubsubs_fsub[0].handle_talk(msg_0)
     msg_1 = make_pubsub_msg(
@@ -246,10 +236,7 @@ async def test_handle_talk(pubsubs_fsub):
         seqno=b"\x11" * 8,
     )
     await pubsubs_fsub[0].handle_talk(msg_1)
-    assert (
-        len(pubsubs_fsub[0].my_topics) == 1
-        and sub == pubsubs_fsub[0].my_topics[TESTING_TOPIC]
-    )
+    assert len(pubsubs_fsub[0].my_topics) == 1 and sub == pubsubs_fsub[0].my_topics[TESTING_TOPIC]
     assert sub.qsize() == 1
     assert (await sub.get()) == msg_0
 
@@ -283,9 +270,7 @@ async def test_publish(pubsubs_fsub, monkeypatch):
     await pubsubs_fsub[0].publish(TESTING_TOPIC, TESTING_DATA)
 
     assert len(msgs) == 2, "`push_msg` should be called every time `publish` is called"
-    assert (msg_forwarders[0] == msg_forwarders[1]) and (
-        msg_forwarders[1] == pubsubs_fsub[0].my_id
-    )
+    assert (msg_forwarders[0] == msg_forwarders[1]) and (msg_forwarders[1] == pubsubs_fsub[0].my_id)
     assert msgs[0].seqno != msgs[1].seqno, "`seqno` should be different every time"