def test_save_svg_button(self):
assembly = HTMLVisualizationAssembly(self.make_adapter(),
save_svg_button=True)
html = assembly.to_html()
self.assertEqual(assembly._call_build_visualization_in_javascript(),
self.get_params({11: 'true'}))
self.assertFalse('<!-- INSERT SCRIPT -->' in html)
def test_show_neutral(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(visualization_data)
._call_build_visualization_in_javascript()),
self.get_params({19: 'false'}))
self.assertEqual((HTMLVisualizationAssembly(visualization_data, show_neutral=True)
._call_build_visualization_in_javascript()),
self.get_params({19: 'true'}))
def test_max_snippets(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(visualization_data,
height_in_pixels=60,
width_in_pixels=1000,
max_snippets=None)
._call_build_visualization_in_javascript()),
self.get_params({0: '1000', 1: '60'}))
self.assertEqual((HTMLVisualizationAssembly(visualization_data,
height_in_pixels=60,
width_in_pixels=1000,
max_snippets=100)
._call_build_visualization_in_javascript()),
self.get_params({0: '1000', 1: '60', 2: '100'}))
def test_height_width_nondefault(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(visualization_data, width_in_pixels=1000)
._call_build_visualization_in_javascript()),
self.get_params({0: '1000'}))
self.assertEqual((HTMLVisualizationAssembly(visualization_data, height_in_pixels=60)
._call_build_visualization_in_javascript()),
self.get_params({1: '60'}))
self.assertEqual((HTMLVisualizationAssembly(visualization_data,
height_in_pixels=60,
width_in_pixels=1000)
._call_build_visualization_in_javascript()),
self.get_params({0: '1000', 1: '60'}))
def test_full_data(self):
visualization_data = self.make_adapter()
full_data = "customFullDataFunction()"
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
full_data=full_data)._call_build_visualization_in_javascript()),
self.get_params({17: full_data}))
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((HTMLVisualizationAssembly(
visualization_data,
color_func=color_func)._call_build_visualization_in_javascript()),
self.get_params({23: color_func}))
def test_y_axis_labels_over_points(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data, y_axis_labels=[
'Lo', 'Hi'
])._call_build_visualization_in_javascript()),
self.get_params({29: '["Lo", "Hi"]'}))
def test_center_label_over_points(self):
visualization_data = self.make_adapter()
self.assertEqual(
(HTMLVisualizationAssembly(visualization_data,
center_label_over_points=True).
_call_build_visualization_in_javascript()),
self.get_params({27: 'true'}))
def test_topic_model_preview_size(self):
visualization_data = self.make_adapter()
self.assertEqual(
(HTMLVisualizationAssembly(visualization_data,
topic_model_preview_size=20).
_call_build_visualization_in_javascript()),
self.get_params({30: '20'}))
def test_color(self):
visualization_data = self.make_adapter()
self.assertEqual(
(HTMLVisualizationAssembly(visualization_data,
color='d3.interpolatePurples').
_call_build_visualization_in_javascript()),
self.get_params({3: 'd3.interpolatePurples'}))
def test_vertical_lines(self):
visualization_data = self.make_adapter()
params = (HTMLVisualizationAssembly(visualization_data,
vertical_lines=[20, 31])
._call_build_visualization_in_javascript())
self.assertEqual(params,
self.get_params({31: '[20, 31]'}))
def test_x_axis_labels(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data, y_axis_values=[
4, 5, 6
])._call_build_visualization_in_javascript()),
self.get_params({22: "[4, 5, 6]"}))
def test_get_tooltip_content(self):
visualization_data = self.make_adapter()
f = '''(function(x) {return 'Original X: ' + x.ox;})'''
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
get_tooltip_content=f)._call_build_visualization_in_javascript()),
self.get_params({20: f}))
def test_word_vec_use_p_vals(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(visualization_data,
height_in_pixels=60,
width_in_pixels=1000,
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((HTMLVisualizationAssembly(visualization_data,
height_in_pixels=60,
width_in_pixels=1000,
show_characteristic=False)
._call_build_visualization_in_javascript()),
self.get_params({0: '1000', 1: '60', 9: 'false'}))
def test_y_label(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(visualization_data,
height_in_pixels=60,
width_in_pixels=1000,
y_label='y label')
._call_build_visualization_in_javascript()),
self.get_params({0: '1000', 1: '60', 16: '"y label"'}))
def test_vertical_line_x_position(self):
visualization_data = self.make_adapter()
params = (HTMLVisualizationAssembly(visualization_data,
vertical_line_x_position=3).
_call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({33: '3'}))
def make_assembler(self):
visualization_data = self.make_adapter()
assembler = HTMLVisualizationAssembly(visualization_data)
return assembler
def test_full_doc(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
use_full_doc=True)._call_build_visualization_in_javascript()),
self.get_params({5: 'true'}))
def test_horizontal_line_y_position(self):
visualization_data = self.make_adapter()
params = (HTMLVisualizationAssembly(visualization_data,
horizontal_line_y_position=0).
_call_build_visualization_in_javascript())
self.assertEqual(params, self.get_params({32: '0'}))
def test_grey_zero_scores(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
grey_zero_scores=True)._call_build_visualization_in_javascript()),
self.get_params({6: 'true'}))
def test_chinese_mode(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
asian_mode=True)._call_build_visualization_in_javascript()),
self.get_params({7: 'true'}))
def test_do_censor_points(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
do_censor_points=False)._call_build_visualization_in_javascript()),
self.get_params({26: 'false'}))
def test_show_axes(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data,
show_axes=False)._call_build_visualization_in_javascript()),
self.get_params({24: 'false'}))
def test_reverse_sort_scores_for_not_category(self):
visualization_data = self.make_adapter()
self.assertEqual((HTMLVisualizationAssembly(
visualization_data, reverse_sort_scores_for_not_category=False).
_call_build_visualization_in_javascript()),
self.get_params({12: 'false'}))
