Last active
April 3, 2020 22:45
-
-
Save GenevieveBuckley/c5f39296836c5a790c17410b625626a1 to your computer and use it in GitHub Desktop.
Screenshot introspection tests
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| def test_screenshot_introspection(viewer_factory): | |
| view, viewer = viewer_factory(show=True) | |
| from skimage.data import camera | |
| _ = viewer.add_image(camera()) | |
| screenshot = viewer.screenshot() | |
| # the screenshot should NOT be completely black | |
| assert not (screenshot == np.array([0, 0, 0, 255])).all() | |
| def test_really_big_2d_image(viewer_factory): | |
| view, viewer = viewer_factory(show=True) | |
| shape = (6898, 9946) | |
| data = np.ones(shape) | |
| _ = viewer.add_image(data) | |
| screenshot = viewer.screenshot() | |
| # Screenshot pixel coordinates to test (in format: row, column) | |
| center_coord = np.round(np.array(screenshot.shape[:2]) / 2).astype(np.int) | |
| top_left_corner = [40, 36] | |
| top_right_corner = [40, 763] | |
| bottom_left_corner = [screenshot.shape[0] - 40, 36] | |
| bottom_right_corner = [screenshot.shape[0] - 40, 763] | |
| expected_value = np.array([255, 255, 255, 255]) # white pixel value | |
| assert all(screenshot[center_coord[0], center_coord[1]] == expected_value) | |
| assert all(screenshot[top_left_corner[0], top_left_corner[1]] == expected_value) | |
| assert all(screenshot[top_right_corner[0], top_right_corner[1]] == expected_value) | |
| assert all(screenshot[bottom_left_corner[0], bottom_left_corner[1]] == expected_value) | |
| assert all(screenshot[bottom_right_corner[0], bottom_right_corner[1]] == expected_value) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment