def test_save_svg_button(self):
scatterplot_structure = ScatterplotStructure(self.make_adapter(), save_svg_button=True)
assembly = BasicHTMLFromScatterplotStructure(scatterplot_structure)
html = assembly.to_html()
self.assertEqual(scatterplot_structure.call_build_visualization_in_javascript(),
self.get_params({11: 'true'}))
self.assertFalse('<!-- INSERT SCRIPT -->' in html)
def test_x_label(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
width_in_pixels=1000,
height_in_pixels=60,
x_label='x label').call_build_visualization_in_javascript()),
self.get_params({
0: '1000',
1: '60',
15: '"x label"'
}))
def test_word_vec_use_p_vals(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
width_in_pixels=1000,
height_in_pixels=60,
word_vec_use_p_vals=True).call_build_visualization_in_javascript()
), self.get_params({
0: '1000',
1: '60',
10: 'true'
}))
def test_show_characteristic(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(visualization_data,
width_in_pixels=1000,
height_in_pixels=60,
show_characteristic=False).
call_build_visualization_in_javascript()),
self.get_params({
0: '1000',
1: '60',
9: 'false'
}))
def test_max_snippets(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
width_in_pixels=1000,
height_in_pixels=60,
max_snippets=None).call_build_visualization_in_javascript()),
self.get_params({
0: '1000',
1: '60'
}))
self.assertEqual((ScatterplotStructure(
visualization_data,
width_in_pixels=1000,
height_in_pixels=60,
max_snippets=100).call_build_visualization_in_javascript()),
self.get_params({
0: '1000',
1: '60',
2: '100'
}))
def test_header_sorting_algos(self):
visualization_data = self.make_adapter()
header_sorting_algos = {
'upper': '(function(a, b) {return b.s - a.s})',
'lower': '(function(a, b) {return a.s - b.s})'
}
params = (ScatterplotStructure(
visualization_data, header_sorting_algos=header_sorting_algos).
call_build_visualization_in_javascript())
self.assertEqual(
params,
self.get_params({
54:
'''{"lower": (function(a, b) {return a.s - b.s}), "upper": (function(a, b) {return b.s - a.s})}'''
}))
def test_header_names(self):
visualization_data = self.make_adapter()
header_names = {
'upper': 'Upper Header Name',
'lower': 'Lower Header Name'
}
params = (ScatterplotStructure(visualization_data,
header_names=header_names).
call_build_visualization_in_javascript())
self.assertEqual(
params,
self.get_params({
53:
'''{"upper": "Upper Header Name", "lower": "Lower Header Name"}'''
}))
def test_y_axis_values_format(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(visualization_data,
y_axis_values_format=".5f").
call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({42: '".5f"'}))
def test_include_all_contexts(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(visualization_data,
include_all_contexts=True).
call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({39: 'true'}))
def test_show_cross_axes(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(
visualization_data,
show_cross_axes=False).call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({36: 'false'}))
def test_vertical_line_x_position(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(visualization_data,
vertical_line_x_position=3).
call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({33: '3'}))
def test_vertical_lines(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(
visualization_data,
vertical_lines=[20, 31]).call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({31: '[20, 31]'}))
def test_center_label_over_points(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(visualization_data,
center_label_over_points=True).
call_build_visualization_in_javascript()),
self.get_params({27: 'true'}))
def test_show_extra(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
show_extra=True).call_build_visualization_in_javascript()),
self.get_params({25: 'true'}))
def test_color_func(self):
visualization_data = self.make_adapter()
color_func = 'function colorFunc(d) {var c = d3.hsl(d3.interpolateRdYlBu(d.x)); c.s *= d.y; return c;}'
self.assertEqual((ScatterplotStructure(visualization_data, color_func=color_func)
.call_build_visualization_in_javascript()),
self.get_params({23: color_func}))
def test_x_axis_labels(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(visualization_data, y_axis_values=[4, 5, 6])
.call_build_visualization_in_javascript()),
self.get_params({22: "[4, 5, 6]"}))
def make_assembler(self):
scatterplot_structure = ScatterplotStructure(self.make_adapter())
return BasicHTMLFromScatterplotStructure(scatterplot_structure)
def test_y_axis_labels_over_points(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(visualization_data, y_axis_labels=['Lo', 'Hi'])
.call_build_visualization_in_javascript()),
self.get_params({29: '["Lo", "Hi"]'}))
def test_do_censor_points(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
do_censor_points=False).call_build_visualization_in_javascript()),
self.get_params({26: 'false'}))
def test_alternative_term_func(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(visualization_data,
alternative_term_func='(function(termDict) {return true;})')
.call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({38: '(function(termDict) {return true;})'}))
def test_topic_model_preview_size(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(visualization_data,
topic_model_preview_size=20).
call_build_visualization_in_javascript()),
self.get_params({30: '20'}))
def test_height_width_default(self):
# assembler = self.make_assembler()
scatterplot_structure = ScatterplotStructure(self.make_adapter())
self.assertEqual(
scatterplot_structure.call_build_visualization_in_javascript(),
self.get_params())
def test_horizontal_line_y_position(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(visualization_data,
horizontal_line_y_position=0).
call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({32: '0'}))
def test_color(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(visualization_data,
color='d3.interpolatePurples').
call_build_visualization_in_javascript()),
self.get_params({3: 'd3.interpolatePurples'}))
def test_unifed_context(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(
visualization_data,
unified_context=True).call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({34: 'true'}))
def test_full_doc(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
use_full_doc=True).call_build_visualization_in_javascript()),
self.get_params({5: 'true'}))
def test_div_name(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(
visualization_data,
div_name='divvydivvy').call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({37: '"divvydivvy"'}))
def test_grey_zero_scores(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
grey_zero_scores=True).call_build_visualization_in_javascript()),
self.get_params({6: 'true'}))
def test_show_axes_and_cross_hairs(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(visualization_data,
show_axes_and_cross_hairs=True).
call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({40: 'true'}))
def test_chinese_mode(self):
visualization_data = self.make_adapter()
self.assertEqual((ScatterplotStructure(
visualization_data,
asian_mode=True).call_build_visualization_in_javascript()),
self.get_params({7: 'true'}))
def test_match_full_line(self):
visualization_data = self.make_adapter()
params = (ScatterplotStructure(
visualization_data,
match_full_line=True).call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({43: 'true'}))
def test_height_width_default(self):
#assembler = self.make_assembler()
scatterplot_structure = ScatterplotStructure(self.make_adapter())
self.assertEqual(scatterplot_structure.call_build_visualization_in_javascript(), self.get_params())
