Unify docker usage in tests

pyspark-notebook/test/test_spark.py

@@ -10,11 +10,9 @@ LOGGER = logging.getLogger(__name__)
 
 def test_spark_shell(container: TrackedContainer) -> None:
     """Checking if Spark (spark-shell) is running properly"""
-    c = container.run(
+    logs = container.run_and_wait(
+        timeout=60,
         tty=True,
         command=["start.sh", "bash", "-c", 'spark-shell <<< "1+1"'],
     )
-    c.wait(timeout=60)
-    logs = c.logs(stdout=True).decode("utf-8")
-    LOGGER.debug(logs)
     assert "res0: Int = 2" in logs, "spark-shell does not work"