def _setup():
    global client_manager, client_factory, controller

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

    client_manager = MagicMock()
    client_factory = MagicMock()
    ipython_display = MagicMock()
    controller = SparkController(ipython_display)
    controller.client_manager = client_manager
    controller.client_factory = client_factory