def test_display_plot():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_storytelling_slope_plot:test_display_plot
"""
source_code = """
from plot_playground.tests.test_storytelling_slope_plot import display_test_plot
display_test_plot()
"""
jupyter_helper.update_ipynb_test_source_code(source_code=source_code)
jupyter_helper.open_test_jupyter_note_book()
jupyter_helper.run_test_code(sleep_seconds=10)
jupyter_helper.hide_header()
jupyter_helper.hide_input_cell()
svg_elem = selenium_helper.driver.find_element_by_id(
settings.TEST_SVG_ELEM_ID)
selenium_helper.save_target_elem_screenshot(target_elem=svg_elem)
selenium_helper.exit_webdriver()
expected_img_path = img_helper.get_test_expected_img_path(
file_name='slope_plot_display_plot')
similarity = img_helper.compare_img_hist(
img_path_1=selenium_helper.DEFAULT_TEST_IMG_PATH,
img_path_2=expected_img_path)
assert_greater_equal(similarity, 0.99)
plot_meta = display_test_plot()
assert_true(isinstance(plot_meta, d3_helper.PlotMeta))
def test__assert_only_one_output_cell_exists():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_jupyter_helper:test__assert_only_one_output_cell_exists
"""
jupyter_helper.update_ipynb_test_source_code(source_code='print(1)')
jupyter_helper.open_test_jupyter_note_book()
jupyter_helper.run_test_code()
jupyter_helper._assert_only_one_output_cell_exists()
def test_get_test_code_text_output():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_jupyter_helper:test_get_test_code_text_output
"""
jupyter_helper.update_ipynb_test_source_code(source_code='print(100)')
jupyter_helper.open_test_jupyter_note_book()
jupyter_helper.run_test_code()
text_output = jupyter_helper.get_test_code_text_output()
assert_equal(text_output, '100')
def test_empty_test_ipynb_code_cell():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_jupyter_helper:test_empty_test_ipynb_code_cell
"""
jupyter_helper.update_ipynb_test_source_code(source_code='print(1)')
jupyter_helper.empty_test_ipynb_code_cell()
jupyter_helper.open_test_jupyter_note_book()
jupyter_helper.run_test_code()
assert_false(jupyter_helper.output_text_cell_exists())
def test_update_ipynb_test_source_code():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_jupyter_helper:test_update_ipynb_test_source_code
"""
jupyter_helper.update_ipynb_test_source_code(
source_code='print(100)\nprint(200)')
jupyter_helper.open_test_jupyter_note_book()
jupyter_helper.run_test_code()
text_output = jupyter_helper.get_test_code_text_output()
is_in = '100' in text_output
assert_true(is_in)
is_in = '200' in text_output
assert_true(is_in)
def test_display_plot():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_linux_stats_plot:test_display_plot
"""
source_code = """
from plot_playground.tests.test_linux_stats_plot import display_test_plot
display_test_plot()
"""
jupyter_helper.update_ipynb_test_source_code(source_code=source_code)
jupyter_helper.open_test_jupyter_note_book()
jupyter_helper.run_test_code(sleep_seconds=10)
jupyter_helper.hide_header()
jupyter_helper.hide_input_cell()
selenium_helper.driver.set_window_size(width=1400, height=1300)
count = 0
while True:
try:
svg_elem = selenium_helper.driver.find_element_by_id(
settings.TEST_SVG_ELEM_ID)
break
except Exception:
count += 1
if count > 5:
break
time.sleep(3)
continue
selenium_helper.save_target_elem_screenshot(target_elem=svg_elem)
expected_img_path = img_helper.get_test_expected_img_path(
file_name='stats_linux_stats_plot_display_plot')
similarity = img_helper.compare_img_hist(
img_path_1=selenium_helper.DEFAULT_TEST_IMG_PATH,
img_path_2=expected_img_path)
assert_greater(similarity, 0.8)
selenium_helper.exit_webdriver()
plot_meta = display_test_plot()
assert_true(isinstance(plot_meta, d3_helper.PlotMeta))
jupyter_helper.empty_test_ipynb_code_cell()
def test_exec_d3_js_script_on_jupyter():
"""
Test Command
------------
$ python run_tests.py --module_name plot_playground.tests.test_d3_helper:test_exec_d3_js_script_on_jupyter
"""
script_str = read_jupyter_test_python_script(
script_file_name='exec_d3_js_script_on_jupyter')
jupyter_helper.update_ipynb_test_source_code(source_code=script_str)
jupyter_helper.selenium_helper.driver.refresh()
time.sleep(3)
jupyter_helper.run_test_code(sleep_seconds=3)
driver = selenium_helper.driver
svg_elem = driver.find_element_by_id(settings.TEST_SVG_ELEM_ID)
selenium_helper.save_target_elem_screenshot(target_elem=svg_elem)
expected_img_path = img_helper.get_test_expected_img_path(
file_name='exec_d3_js_script_on_jupyter')
similarity = img_helper.compare_img_hist(
img_path_1=selenium_helper.DEFAULT_TEST_IMG_PATH,
img_path_2=expected_img_path)
assert_equal(similarity, 1.0)