def _setup(): global client_manager, controller, ipython_display client_manager = MagicMock() ipython_display = MagicMock() spark_events = MagicMock() controller = SparkController(ipython_display) controller.session_manager = client_manager controller.spark_events = spark_events