From dc0d31a96ce44b7668acec5947f37f9f13a59af6 Mon Sep 17 00:00:00 2001 From: chenyangkang Date: Fri, 31 Oct 2025 08:58:55 -0500 Subject: [PATCH] update pytest for plot_gif --- tests/test_makegif.py | 48 +++++++++++++++++++++++-------------------- 1 file changed, 26 insertions(+), 22 deletions(-) diff --git a/tests/test_makegif.py b/tests/test_makegif.py index d3d4d5b..d35adc8 100644 --- a/tests/test_makegif.py +++ b/tests/test_makegif.py @@ -1,6 +1,6 @@ import os import shutil - +import pytest import numpy as np import pandas as pd @@ -73,32 +73,36 @@ def test_make_gif_changing_ranges(): shutil.rmtree(tmp_dir) -def test_make_gif_plot_political_boundary(): +def test_make_gif_plot_political_boundary_raise_error(): tmp_dir = "./stemflow_test_make_gif3" if not os.path.exists(tmp_dir): os.mkdir(tmp_dir) - make_sample_gif( - fake_data, - os.path.join(tmp_dir, "FTR_IPT_dat_changing_ranges.gif"), - col="dat", - log_scale=False, - Spatio1="x", - Spatio2="y", - Temporal1="DOY", - figsize=(18, 9), - xlims=None, - ylims=None, - grid=True, - xtick_interval=None, - ytick_interval=None, - lng_size=10, - lat_size=10, - dpi=100, - fps=10, - political_boundary='both' - ) + + with pytest.raises(ValueError): + + make_sample_gif( + fake_data, + os.path.join(tmp_dir, "FTR_IPT_dat_changing_ranges.gif"), + col="dat", + log_scale=False, + Spatio1="x", + Spatio2="y", + Temporal1="DOY", + figsize=(18, 9), + xlims=None, + ylims=None, + grid=True, + xtick_interval=None, + ytick_interval=None, + lng_size=10, + lat_size=10, + dpi=100, + fps=10, + continental_boundary=False, + political_boundary='both' + ) shutil.rmtree(tmp_dir)