def test_emit_graph_render_event_when_registered():
conf.override(conf.events_handler.__name__, events.handler)
event_name = GRAPH_RENDER_EVENT
graph_type = 'Bar'
kwargs_list = [(INSTANCE_ID, get_instance_id()), (EVENT_NAME, event_name),
(TIMESTAMP, time_stamp), (GRAPH_TYPE, graph_type)]
events.emit_graph_render_event(graph_type)
events.get_utc_date_time.assert_called_with()
assert not events.handler.handle_event.called
def _register_auto_viz(self):
from sparkmagic.utils.sparkevents import get_spark_events_handler
import autovizwidget.utils.configuration as c
handler = get_spark_events_handler()
c.override("events_handler", handler)
register_auto_viz_code = """from autovizwidget.widget.utils import display_dataframe
ip = get_ipython()
ip.display_formatter.ipython_display_formatter.for_type_by_name('pandas.core.frame', 'DataFrame', display_dataframe)"""
self._execute_cell(register_auto_viz_code, True, False, shutdown_if_error=True,
log_if_error="Failed to register auto viz for notebook.")
self.logger.debug("Registered auto viz.")
def _register_auto_viz(self):
from sparkmagic.utils.sparkevents import get_spark_events_handler
import autovizwidget.utils.configuration as c
handler = get_spark_events_handler()
c.override("events_handler", handler)
register_auto_viz_code = """from autovizwidget.widget.utils import display_dataframe
ip = get_ipython()
ip.display_formatter.ipython_display_formatter.for_type_by_name('pandas.core.frame', 'DataFrame', display_dataframe)"""
self._execute_cell(register_auto_viz_code, True, False, shutdown_if_error=True,
log_if_error="Failed to register auto viz for notebook.")
self.logger.debug("Registered auto viz.")
def test_emit_graph_render_event_when_registered():
conf.override(conf.events_handler.__name__, events.handler)
event_name = GRAPH_RENDER_EVENT
graph_type = 'Bar'
kwargs_list = [(INSTANCE_ID, get_instance_id()),
(EVENT_NAME, event_name),
(TIMESTAMP, time_stamp),
(GRAPH_TYPE, graph_type)]
events.emit_graph_render_event(graph_type)
events.get_utc_date_time.assert_called_with()
assert not events.handler.handle_event.called