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_open_test_jupyter_note_book(): """ Test Command ------------ $ python run_tests.py --module_name plot_playground.tests.test_jupyter_helper:test_open_test_jupyter_note_book """ jupyter_helper.open_test_jupyter_note_book() assert_equal( selenium_helper.driver.title, jupyter_helper.TEST_JUPYTER_NOTE_NAME, 'could not open the Jupyter notebook for testing. Please check that the notebook is properly placed in the target path.' ) selenium_helper.exit_webdriver()
def test__assert_current_page_is_test_notebook(): """ Test Command ------------ $ python run_tests.py --module_name plot_playground.tests.test_jupyter_helper:test__assert_current_page_is_test_notebook """ selenium_helper.exit_webdriver() assert_raises(Exception, jupyter_helper._assert_current_page_is_test_notebook) driver = selenium_helper.start_webdriver() driver.get('https://www.google.com/') assert_raises(Exception, jupyter_helper._assert_current_page_is_test_notebook) jupyter_helper.open_test_jupyter_note_book() jupyter_helper._assert_current_page_is_test_notebook()
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 teardown(): selenium_helper.exit_webdriver() jupyter_helper.empty_test_ipynb_code_cell()
def test_exit_webdriver(): driver = selenium_helper.start_webdriver() assert_not_equal(driver, None) selenium_helper.exit_webdriver() assert_equal(selenium_helper.driver, None)