From 774bd22c444c46b4970af108bed426829651d390 Mon Sep 17 00:00:00 2001 From: Will Schlitzer Date: Thu, 28 Jan 2021 20:39:17 +0000 Subject: [PATCH 1/2] Add test for savefig invalid file extension --- pygmt/tests/test_figure.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/pygmt/tests/test_figure.py b/pygmt/tests/test_figure.py index f4ef5e02ee6..dd248cb4fa9 100644 --- a/pygmt/tests/test_figure.py +++ b/pygmt/tests/test_figure.py @@ -65,6 +65,19 @@ def test_figure_savefig_exists(): os.remove(fname) +def test_figure_savefig_unknown_extension(): + """ + Check that an error is raised when an unknown extension is passed. + """ + fig = Figure() + fig.basemap(region="10/70/-300/800", projection="X3i/5i", frame="af") + prefix = "test_figure_savefig_transparent" + fmt = "123" + fname = ".".join([prefix, fmt]) + with pytest.raises(GMTInvalidInput): + fig.savefig(fname) + + def test_figure_savefig_transparent(): """ Check if fails when transparency is not supported. From 746d9613f0aa7fa71b01746dd154dc86ceb3fba7 Mon Sep 17 00:00:00 2001 From: Will Schlitzer Date: Thu, 28 Jan 2021 20:44:24 +0000 Subject: [PATCH 2/2] Add test for invalid show method passed --- pygmt/tests/test_figure.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/pygmt/tests/test_figure.py b/pygmt/tests/test_figure.py index dd248cb4fa9..99cb7d8e73e 100644 --- a/pygmt/tests/test_figure.py +++ b/pygmt/tests/test_figure.py @@ -71,8 +71,8 @@ def test_figure_savefig_unknown_extension(): """ fig = Figure() fig.basemap(region="10/70/-300/800", projection="X3i/5i", frame="af") - prefix = "test_figure_savefig_transparent" - fmt = "123" + prefix = "test_figure_savefig_unknown_extension" + fmt = "test" fname = ".".join([prefix, fmt]) with pytest.raises(GMTInvalidInput): fig.savefig(fname) @@ -163,3 +163,14 @@ def test_shift_origin(): fig.shift_origin(xshift="-4i", yshift="6i") fig.basemap(region="10/70/-300/300", projection="X3i/5i", frame="af") return fig + + +def test_figure_show_invalid_method(): + """ + Test to check if an error is raised when an invalid method is passed to + show. + """ + fig = Figure() + fig.basemap(region="10/70/-300/800", projection="X3i/5i", frame="af") + with pytest.raises(GMTInvalidInput): + fig.show(method="test")