From 9b0bfb3b884e624b46be2191a851357fcfefa135 Mon Sep 17 00:00:00 2001 From: kushalkolar Date: Fri, 30 Jun 2023 16:47:39 -0400 Subject: [PATCH] start logic for nb screenshot tests --- examples/notebooks/nb_test_utils.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 examples/notebooks/nb_test_utils.py diff --git a/examples/notebooks/nb_test_utils.py b/examples/notebooks/nb_test_utils.py new file mode 100644 index 000000000..576d82806 --- /dev/null +++ b/examples/notebooks/nb_test_utils.py @@ -0,0 +1,27 @@ +import os + +import imageio as iio + +from fastplotlib.layouts._base import PlotArea + + +def test_plot(name, plot: PlotArea): + if os.environ["REGENERATE_SCREENSHOTS"] == "1": + snapshot = plot.canvas.snapshot() + regenerate_screenshot(snapshot.data) + + assert_screenshot_equal(name, snapshot.data) + + +def regenerate_screenshot(name, data): + pass + + +def assert_screenshot_equal(name, data): + # do stuff + if not is_similar: + update_screenshot_diff(name, data) + + +def update_screenshot_diff(name, data): + pass