fix(lint): functional tests

Author: prajwolrg

functional-tests/tests/prover/prover_checkpoint_latest.py

@@ -1,5 +1,3 @@
-import time
-
 import flexitest
 
 from envs import testenv
@@ -29,5 +27,7 @@ def main(self, ctx: flexitest.RunContext):
         assert task_id is not None
 
         time_out = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=time_out)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=time_out
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_checkpoint_manual.py

@@ -1,15 +1,13 @@
-import time
-
-from bitcoinlib.services.bitcoind import BitcoindClient
 import flexitest
+from bitcoinlib.services.bitcoind import BitcoindClient
 
 from envs import testenv
-from utils import wait_for_proof_with_time_out, wait_until, bytes_to_big_endian, cl_slot_to_block_id
+from utils import bytes_to_big_endian, cl_slot_to_block_id, wait_for_proof_with_time_out, wait_until
 
 CHECKPOINT_PROVER_PARAMS = {
     "checkpoint_idx": 1,
-    "l1_range": (1, 2),
-    "l2_range": (1, 2),
+    "l1_range": (1, 1),
+    "l2_range": (1, 1),
 }
 
 
@@ -54,14 +52,18 @@ def main(self, ctx: flexitest.RunContext):
         l2_end_block_commitment = {"slot": slot, "blkid": block_id}
 
         task_ids = prover_client_rpc.dev_strata_proveCheckpointRaw(
-            CHECKPOINT_PROVER_PARAMS["checkpoint_idx"], (l1_start_block_commitment, l1_end_block_commitment), (l2_start_block_commitment, l2_end_block_commitment)
+            CHECKPOINT_PROVER_PARAMS["checkpoint_idx"],
+            (l1_start_block_commitment, l1_end_block_commitment),
+            (l2_start_block_commitment, l2_end_block_commitment),
         )
         self.debug(f"got the task ids: {task_ids}")
         assert task_ids is not None
 
         time_out = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_ids[0], time_out=time_out)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_ids[0], time_out=time_out
+        )
 
         # Proof generation is expected to fail because the range will not match
         # CL STF Proof will fail, which in turns fails the checkpoint proof
-        assert(not is_proof_generation_completed)
+        assert not is_proof_generation_completed

functional-tests/tests/prover/prover_cl_dispatch.py

@@ -1,5 +1,3 @@
-import time
-
 import flexitest
 
 from envs import testenv
@@ -46,5 +44,7 @@ def main(self, ctx: flexitest.RunContext):
         assert task_id is not None
 
         time_out = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=time_out)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=time_out
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_client_restart.py

@@ -67,5 +67,7 @@ def prove_latest_checkpoint(self, prover_client_rpc):
         assert task_id is not None
 
         time_out = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=time_out)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=time_out
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_el_deposit_withdraw.py

@@ -161,5 +161,7 @@ def test_checkpoint(self, l1_block, l2_block, prover_client_rpc, rethrpc):
         self.debug(f"using task id: {task_id}")
         assert task_id is not None
 
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=30)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=30
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_el_dispatch.py

@@ -64,5 +64,7 @@ def main(self, ctx: flexitest.RunContext):
         task_id = task_ids[0]
         self.debug(f"Using task ID: {task_id}")
 
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=30)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=30
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_el_precompiles.py

@@ -66,5 +66,7 @@ def main(self, ctx: flexitest.RunContext):
         task_id = task_ids[0]
         self.debug(f"Using task ID: {task_id}")
 
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=30)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=30
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_l1_batch_dispatch.py

@@ -1,5 +1,3 @@
-import time
-
 import flexitest
 from bitcoinlib.services.bitcoind import BitcoindClient
 
@@ -46,5 +44,7 @@ def main(self, ctx: flexitest.RunContext):
         assert task_id is not None
 
         proof_timeout_seconds = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=proof_timeout_seconds)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=proof_timeout_seconds
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_l1_dispatch.py

@@ -1,5 +1,3 @@
-import time
-
 import flexitest
 from bitcoinlib.services.bitcoind import BitcoindClient
 
@@ -39,5 +37,7 @@ def main(self, ctx: flexitest.RunContext):
         assert task_id is not None
 
         time_out = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=time_out)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=time_out
+        )
+        assert is_proof_generation_completed

functional-tests/tests/prover/prover_schnorr_precompile.py

@@ -62,5 +62,7 @@ def main(self, ctx: flexitest.RunContext):
         assert task_id is not None
 
         time_out = 30
-        is_proof_generation_completed = wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=time_out)
-        assert(is_proof_generation_completed)
+        is_proof_generation_completed = wait_for_proof_with_time_out(
+            prover_client_rpc, task_id, time_out=time_out
+        )
+        assert is_proof_generation_completed

functional-tests/utils/utils.py

@@ -450,14 +450,14 @@ def check_already_sent_proof(seqrpc, sent_batch: int):
         raise AssertionError("Expected rpc error")
 
 
-def wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=3600) -> bool :
+def wait_for_proof_with_time_out(prover_client_rpc, task_id, time_out=3600) -> bool:
     """
     Waits for a proof task to complete/fail within a specified timeout period.
 
     This function continuously polls the status of a proof task identified by `task_id` using
     the `prover_client_rpc` client. It checks the status every 2 seconds and waits until the
-    proof task status is either "Completed" where it returns True, or "Failed" where it return False.
-    If the specified `time_out` (in seconds) is reached, it throws TimeoutError.
+    proof task status is either "Completed" where it returns True, or "Failed" where it return
+    False. If the specified `time_out` (in seconds) is reached, it throws TimeoutError.
     """
 
     start_time = time.time()