more tests

Author: mathause

mplotutils/test/test_colorbar_utils.py

@@ -5,7 +5,8 @@
  resize_colorbar_horz,
  _parse_shift_shrink)
 
-
+import matplotlib.pyplot as plt
+import numpy as np
 
 
 def test_parse_shift_shrink():
@@ -32,3 +33,200 @@ def test_parse_shift_shrink():
 
  with raises(AssertionError):
  _parse_shift_shrink(0, 1.1)
+
+
+def _easy_cbar_vert(**kwargs):
+
+ f, ax = plt.subplots()
+ 
+ f.subplots_adjust(left=0, bottom=0, right=0.8, top=1)
+
+ # simplest 'mappable'
+ h = ax.pcolormesh([[0, 1]])
+ 
+ # create colorbar
+ cbax = f.add_axes([0, 0, 0.1, 0.1])
+
+ cbar = plt.colorbar(h, orientation='vertical', cax=cbax)
+
+ func = resize_colorbar_vert(cbax, ax, **kwargs)
+ f.canvas.mpl_connect('draw_event', func)
+
+ f.canvas.draw()
+
+ return f, cbar
+
+
+def test_resize_colorbar_vert():
+
+
+ # test pad=0, size=0.2
+ f, cbar = _easy_cbar_vert(size=0.2, pad=0)
+ 
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.8, 0, 0.2, 1.]
+
+ np.testing.assert_allclose(res, exp)
+
+ # -----------------------------------------------------------
+
+ f.subplots_adjust(left=0, bottom=0.1, right=0.8, top=0.9)
+ f.canvas.draw()
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.8, 0.1, 0.2, 0.8]
+
+ np.testing.assert_allclose(res, exp)
+ 
+ # ===========================================================
+
+ # pad=0.05, size=0.1
+
+ f, cbar = _easy_cbar_vert(size=0.1, pad=0.05)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.85, 0, 0.1, 1.]
+
+ np.testing.assert_allclose(res, exp)
+
+ # ===========================================================
+
+ # shift='symmetric', shrink=0.1 
+ # --> colorbar is 10 % smaller, and centered
+
+ f, cbar = _easy_cbar_vert(size=0.2, pad=0., shrink=0.1)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.8, 0.05, 0.2, 0.9]
+
+ np.testing.assert_allclose(res, exp)
+
+ # ===========================================================
+
+ # shift=0., shrink=0.1
+ # --> colorbar is 10 % smaller, and aligned with the bottom
+
+ f, cbar = _easy_cbar_vert(size=0.2, pad=0., shrink=0.1, shift=0.)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.8, 0.0, 0.2, 0.9]
+
+ np.testing.assert_allclose(res, exp)
+
+ # ===========================================================
+ 
+ # shift=0.1, shrink=None
+ # --> colorbar is 10 % smaller, and aligned with the top
+
+ f, cbar = _easy_cbar_vert(size=0.2, pad=0., shrink=None, shift=0.1)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.8, 0.1, 0.2, 0.9]
+
+ np.testing.assert_allclose(res, exp)
+
+# =============================================================================
+# =============================================================================
+
+def _easy_cbar_horz(**kwargs):
+
+ f, ax = plt.subplots()
+ 
+ f.subplots_adjust(left=0, bottom=0.2, right=1, top=1)
+
+ # simplest 'mappable'
+ h = ax.pcolormesh([[0, 1]])
+ 
+ # create colorbar
+ cbax = f.add_axes([0, 0, 0.1, 0.1])
+
+ cbar = plt.colorbar(h, orientation='horizontal', cax=cbax)
+
+ func = resize_colorbar_horz(cbax, ax, **kwargs)
+ f.canvas.mpl_connect('draw_event', func)
+
+ f.canvas.draw()
+
+ return f, cbar
+
+
+def test_resize_colorbar_horz():
+
+
+ # test pad=0, size=0.2
+ f, cbar = _easy_cbar_horz(size=0.2, pad=0)
+ 
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0, 0, 1, 0.2]
+
+ # adding atol because else 5e-17 != 0
+ np.testing.assert_allclose(res, exp, atol=1e-08)
+
+ # -----------------------------------------------------------
+
+ f.subplots_adjust(left=0.1, bottom=0.2, right=0.9, top=1)
+ f.canvas.draw()
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.1, 0, 0.8, 0.2]
+
+ np.testing.assert_allclose(res, exp, atol=1e-08)
+ 
+ # ===========================================================
+
+ # pad=0.05, size=0.1
+
+ f, cbar = _easy_cbar_horz(size=0.1, pad=0.05)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.0, 0.05, 1, 0.1]
+
+ np.testing.assert_allclose(res, exp, atol=1e-08)
+
+ # ===========================================================
+
+ # shift='symmetric', shrink=0.1 
+ # --> colorbar is 10 % smaller, and centered
+
+ f, cbar = _easy_cbar_horz(size=0.2, pad=0., shrink=0.1)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.05, 0, 0.9, 0.2]
+
+ np.testing.assert_allclose(res, exp, atol=1e-08)
+
+ # ===========================================================
+
+ # shift=0., shrink=0.1
+ # --> colorbar is 10 % smaller, and aligned with lhs
+
+ f, cbar = _easy_cbar_horz(size=0.2, pad=0., shrink=0.1, shift=0.)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.0, 0, 0.9, 0.2]
+
+ np.testing.assert_allclose(res, exp, atol=1e-08)
+
+ # ===========================================================
+ 
+ # shift=0.1, shrink=None
+ # --> colorbar is 10 % smaller, and aligned with rhs
+
+ f, cbar = _easy_cbar_horz(size=0.2, pad=0., shrink=None, shift=0.1)
+
+ pos = cbar.ax.get_position()
+ res = [pos.x0, pos.y0, pos.width, pos.height]
+ exp = [0.1, 0, 0.9, 0.2]
+
+ np.testing.assert_allclose(res, exp, atol=1e-08)

mplotutils/test/test_cyclic_dataarray.py

@@ -0,0 +1,30 @@
+
+import numpy as np
+import xarray as xr
+
+from mplotutils.cartopy_utils import (cyclic_dataarray)
+
+
+def test_cyclic_dataarray():
+
+ data = xr.DataArray([[1, 2, 3], [4, 5, 6]],
+ coords={'x': [1, 2], 'y': range(3)},
+ dims=['x', 'y'])
+
+ res = cyclic_dataarray(data, 'y')
+ 
+ expected_data = np.asarray([[1, 2, 3, 1], [4, 5, 6, 4]])
+ 
+ np.testing.assert_allclose(res, expected_data)
+
+ np.testing.assert_allclose(res.y, [0, 1, 2, 3]) 
+ np.testing.assert_allclose(res.x, [1, 2]) 
+
+
+ # per default use 'lon'
+ data = xr.DataArray([[1, 2, 3], [4, 5, 6]],
+ coords={'x': [1, 2], 'lon': range(3)},
+ dims=['x', 'lon'])
+
+ res = cyclic_dataarray(data)
+ np.testing.assert_allclose(res, expected_data)

mplotutils/test/test_infer_interval_breaks.py

@@ -0,0 +1,97 @@
+import numpy as np
+
+from numpy.testing import assert_array_equal # noqa: F401
+
+from pytest import raises
+
+from mplotutils.cartopy_utils import (infer_interval_breaks, 
+ _infer_interval_breaks)
+
+
+def test__infer_interval_breaks():
+ assert_array_equal([-0.5, 0.5, 1.5], _infer_interval_breaks([0, 1]))
+ assert_array_equal([-0.5, 0.5, 5.0, 9.5, 10.5],
+ _infer_interval_breaks([0, 1, 9, 10]))
+
+ # make a bounded 2D array that we will center and re-infer
+ xref, yref = np.meshgrid(np.arange(6), np.arange(5))
+ cx = (xref[1:, 1:] + xref[:-1, :-1]) / 2
+ cy = (yref[1:, 1:] + yref[:-1, :-1]) / 2
+ x = _infer_interval_breaks(cx, axis=1)
+ x = _infer_interval_breaks(x, axis=0)
+ y = _infer_interval_breaks(cy, axis=1)
+ y = _infer_interval_breaks(y, axis=0)
+ np.testing.assert_allclose(xref, x)
+ np.testing.assert_allclose(yref, y)
+
+ # test that warning is raised for non-monotonic inputs
+ # with raises(ValueError):
+ # _infer_interval_breaks(np.array([0, 2, 1]))
+
+
+def test_infer_interval_breaks():
+ 
+ # 1D
+ lon = np.arange(5, 356, 10)
+ lat = np.arange(-85, 86, 10)
+
+ lon_expected = np.arange(0, 361, 10)
+ lat_expected = np.arange(-90, 91, 10)
+
+ lon_result, lat_result = infer_interval_breaks(lon, lat)
+
+ np.testing.assert_allclose(lon_expected, lon_result)
+ np.testing.assert_allclose(lat_expected, lat_result)
+
+ # 2D, as above
+
+ xref, yref = np.meshgrid(np.arange(6), np.arange(5))
+ cx = (xref[1:, 1:] + xref[:-1, :-1]) / 2
+ cy = (yref[1:, 1:] + yref[:-1, :-1]) / 2
+ 
+ x, y = infer_interval_breaks(cx, cy)
+
+ np.testing.assert_allclose(xref, x)
+ np.testing.assert_allclose(yref, y)
+
+
+def test_infer_interval_breaks_clip():
+ 
+ # no clip
+ lon = np.arange(5, 356, 10)
+ lat = np.arange(-90, 91, 10)
+
+ lon_expected = np.arange(0, 361, 10)
+ lat_expected = np.arange(-95, 96, 10)
+
+ lon_result, lat_result = infer_interval_breaks(lon, lat)
+
+ np.testing.assert_allclose(lon_expected, lon_result)
+ np.testing.assert_allclose(lat_expected, lat_result)
+
+ # clip
+ lat_expected[0] = -90
+ lat_expected[-1] = 90
+
+ lon_result, lat_result = infer_interval_breaks(lon, lat, clip=True)
+
+ np.testing.assert_allclose(lon_expected, lon_result)
+ np.testing.assert_allclose(lat_expected, lat_result)
+
+
+
+ # 2D, as above
+
+
+
+# def test_is_monotonic():
+
+ # _is_monotonic(coord, axis=0):
+ """
+ >>> _is_monotonic(np.array([0, 1, 2]))
+ True
+ >>> _is_monotonic(np.array([2, 1, 0]))
+ True
+ >>> _is_monotonic(np.array([0, 2, 1]))
+ False
+ """

mplotutils/test/test_sample_data_3d.py

@@ -0,0 +1,41 @@
+from mplotutils import sample_data_3d
+
+import numpy as np
+
+def test_data_shape():
+
+ nlons = 10
+ nlats = 20
+ 
+ lon, lat, data = sample_data_3d(nlons, nlats)
+
+ assert len(lon) == nlons
+ assert len(lat) == nlats
+
+ assert data.shape == (nlats, nlons)
+
+ nlons = 5
+ nlats = 10
+
+ lon, lat, data = sample_data_3d(nlons, nlats)
+
+ assert len(lon) == nlons
+ assert len(lat) == nlats
+
+ assert data.shape == (nlats, nlons)
+
+
+def test_lat():
+
+ __, lat, __ = sample_data_3d(36, 9)
+
+ expected_lat = np.arange(-80, 81, 20)
+ assert np.allclose(lat, expected_lat)
+
+
+def test_lon():
+
+ lon, __, __ = sample_data_3d(36, 9)
+
+ expected_lon = np.arange(0, 351, 10)
+ assert np.allclose(lon, expected_lon)