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'}))