def test_convert_to_displayable_dataframe():
bool_df = pd.DataFrame([{
u'bool_col': True,
u'int_col': 0,
u'float_col': 3.0
}, {
u'bool_col': False,
u'int_col': 100,
u'float_col': 0.7
}])
copy_of_df = bool_df.copy()
widget = AutoVizWidget(df,
encoding,
renderer,
ipywidget_factory,
encoding_widget,
ipython_display,
spark_events=spark_events,
testing=True)
result = AutoVizWidget._convert_to_displayable_dataframe(bool_df)
# Ensure original DF not changed
assert_frame_equal(bool_df, copy_of_df)
assert_series_equal(bool_df[u'int_col'], result[u'int_col'])
assert_series_equal(bool_df[u'float_col'], result[u'float_col'])
assert_equals(result.dtypes[u'bool_col'], object)
assert_equals(len(result[u'bool_col']), 2)
assert_equals(result[u'bool_col'][0], 'True')
assert_equals(result[u'bool_col'][1], 'False')
spark_events.emit_graph_render_event.assert_called_once_with(
encoding.chart_type)
def test_on_render_viz():
widget = AutoVizWidget(df,
encoding,
renderer,
ipywidget_factory,
encoding_widget,
ipython_display,
spark_events=spark_events,
testing=True)
# on_render_viz is called in the constructor, so no need to call it here.
output.clear_output.assert_called_once_with()
assert_equals(len(renderer.render.mock_calls), 1)
assert_equals(len(renderer.render.mock_calls[0]), 3)
assert_equals(renderer.render.mock_calls[0][1][1], encoding)
assert_equals(renderer.render.mock_calls[0][1][2], output)
assert_frame_equal(renderer.render.mock_calls[0][1][0], df)
encoding_widget.show_x.assert_called_once_with(True)
encoding_widget.show_y.assert_called_once_with(True)
encoding_widget.show_controls.assert_called_once_with(True)
encoding_widget.show_logarithmic_x_axis.assert_called_once_with(True)
encoding_widget.show_logarithmic_y_axis.assert_called_once_with(True)
spark_events.emit_graph_render_event.assert_called_once_with(
encoding.chart_type)
encoding._chart_type = Encoding.chart_type_scatter
widget.on_render_viz()
assert_equals(len(spark_events.emit_graph_render_event.mock_calls), 2)
assert_equals(spark_events.emit_graph_render_event.call_args,
call(Encoding.chart_type_scatter))
def test_on_render_viz():
widget = AutoVizWidget(df, encoding, renderer, ipywidget_factory,
encoding_widget, ipython_display, spark_events=spark_events, testing=True)
# on_render_viz is called in the constructor, so no need to call it here.
output.clear_output.assert_called_once_with()
assert_equals(len(renderer.render.mock_calls), 1)
assert_equals(len(renderer.render.mock_calls[0]), 3)
assert_equals(renderer.render.mock_calls[0][1][1], encoding)
assert_equals(renderer.render.mock_calls[0][1][2], output)
assert_frame_equal(renderer.render.mock_calls[0][1][0], df)
encoding_widget.show_x.assert_called_once_with(True)
encoding_widget.show_y.assert_called_once_with(True)
encoding_widget.show_controls.assert_called_once_with(True)
encoding_widget.show_logarithmic_x_axis.assert_called_once_with(True)
encoding_widget.show_logarithmic_y_axis.assert_called_once_with(True)
spark_events.emit_graph_render_event.assert_called_once_with(encoding.chart_type)
encoding._chart_type = Encoding.chart_type_scatter
widget.on_render_viz()
assert_equals(len(spark_events.emit_graph_render_event.mock_calls), 2)
assert_equals(spark_events.emit_graph_render_event.call_args, call(Encoding.chart_type_scatter))
def test_convert_to_displayable_dataframe():
bool_df = pd.DataFrame([{
u'bool_col': True,
u'int_col': 0,
u'float_col': 3.0
}, {
u'bool_col': False,
u'int_col': 100,
u'float_col': 0.7
}])
copy_of_df = bool_df.copy()
widget = AutoVizWidget(
df,
encoding,
renderer,
ipywidget_factory,
encoding_widget,
ipython_display,
spark_events=spark_events,
testing=True)
result = AutoVizWidget._convert_to_displayable_dataframe(bool_df)
# Ensure original DF not changed
assert_frame_equal(bool_df, copy_of_df)
assert_series_equal(bool_df[u'int_col'], result[u'int_col'])
assert_series_equal(bool_df[u'float_col'], result[u'float_col'])
assert_equals(result.dtypes[u'bool_col'], object)
assert_equals(len(result[u'bool_col']), 2)
assert_equals(result[u'bool_col'][0], 'True')
assert_equals(result[u'bool_col'][1], 'False')
spark_events.emit_graph_render_event.assert_called_once_with(
encoding.chart_type)
def test_create_viz_types_buttons():
df_single_column = pd.DataFrame([{u'buildingID': 0}])
widget = AutoVizWidget(df_single_column,
encoding,
renderer,
ipywidget_factory,
encoding_widget,
ipython_display,
spark_events=spark_events,
testing=True)
# create_viz_types_buttons is called in the constructor, so no need to call it here.
assert call(description=Encoding.chart_type_table
) in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_pie
) in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_line
) not in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_area
) not in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_bar
) not in ipywidget_factory.get_button.mock_calls
spark_events.emit_graph_render_event.assert_called_once_with(
encoding.chart_type)
widget = AutoVizWidget(df,
encoding,
renderer,
ipywidget_factory,
encoding_widget,
ipython_display,
spark_events=spark_events,
testing=True)
# create_viz_types_buttons is called in the constructor, so no need to call it here.
assert call(description=Encoding.chart_type_table
) in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_pie
) in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_line
) in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_area
) in ipywidget_factory.get_button.mock_calls
assert call(description=Encoding.chart_type_bar
) in ipywidget_factory.get_button.mock_calls
def test_on_render_viz():
widget = AutoVizWidget(df, encoding, renderer, ipywidget_factory,
encoding_widget, ipython_display, testing=True)
# on_render_viz is called in the constructor, so no need to call it here.
output.clear_output.assert_called_once()
renderer.render.assert_called_once_with(df, encoding, output)
encoding_widget.show_x.assert_called_once_with(True)
encoding_widget.show_y.assert_called_once_with(True)
encoding_widget.show_controls.assert_called_once_with(True)
encoding_widget.show_logarithmic_x_axis.assert_called_once_with(True)
encoding_widget.show_logarithmic_y_axis.assert_called_once_with(True)
def test_create_viz_empty_df():
df = pd.DataFrame([])
widget = AutoVizWidget(df,
encoding,
renderer,
ipywidget_factory,
encoding_widget,
ipython_display,
spark_events=spark_events,
testing=True)
ipywidget_factory.get_button.assert_not_called()
ipywidget_factory.get_html.assert_called_once_with("No results.")
ipython_display.display.assert_called_with(
ipywidget_factory.get_html.return_value)
spark_events.emit_graph_render_event.assert_called_once_with(
encoding.chart_type)
