From 061ddf2b68f30a2fe036cb1ad672bd4ff81a0dd4 Mon Sep 17 00:00:00 2001
From: John Omotani <john.omotani@ukaea.uk>
Date: Wed, 5 Feb 2020 17:55:52 +0000
Subject: [PATCH] Don't use private members in tests

---
 xarray/tests/test_plot.py | 27 ++++++++++++++++++---------
 1 file changed, 18 insertions(+), 9 deletions(-)

diff --git a/xarray/tests/test_plot.py b/xarray/tests/test_plot.py
index 26daf417037..330c975b999 100644
--- a/xarray/tests/test_plot.py
+++ b/xarray/tests/test_plot.py
@@ -284,13 +284,16 @@ def test_contourf_cmap_set(self):
         pl = a.plot.contourf(cmap=deepcopy(cmap), vmin=0.1, vmax=0.9)
 
         # check the set_bad color
-        assert pl.cmap._rgba_bad == cmap._rgba_bad
+        assert np.all(
+            pl.cmap(np.ma.masked_invalid([np.nan]))[0]
+            == cmap(np.ma.masked_invalid([np.nan]))[0]
+        )
 
         # check the set_under color
-        assert pl.cmap._rgba_under == cmap._rgba_under
+        assert pl.cmap(-np.inf) == cmap(-np.inf)
 
         # check the set_over color
-        assert pl.cmap._rgba_over == cmap._rgba_over
+        assert pl.cmap(np.inf) == cmap(np.inf)
 
     def test_contourf_cmap_set_with_bad_under_over(self):
         a = DataArray(easy_array((4, 4)), dims=["z", "time"])
@@ -302,27 +305,33 @@ def test_contourf_cmap_set_with_bad_under_over(self):
 
         cmap.set_bad("w")
         # check we actually changed the set_bad color
-        assert cmap._rgba_bad != mpl.cm.viridis._rgba_bad
+        assert np.all(
+            cmap(np.ma.masked_invalid([np.nan]))[0]
+            != mpl.cm.viridis(np.ma.masked_invalid([np.nan]))[0]
+        )
 
         cmap.set_under("r")
         # check we actually changed the set_under color
-        assert cmap._rgba_under != mpl.cm.viridis._rgba_under
+        assert cmap(-np.inf) != mpl.cm.viridis(-np.inf)
 
         cmap.set_over("g")
         # check we actually changed the set_over color
-        assert cmap._rgba_over != mpl.cm.viridis._rgba_over
+        assert cmap(np.inf) != mpl.cm.viridis(-np.inf)
 
         # deepcopy to ensure cmap is not changed by contourf()
         pl = a.plot.contourf(cmap=deepcopy(cmap))
 
         # check the set_bad color has been kept
-        assert pl.cmap._rgba_bad == cmap._rgba_bad
+        assert np.all(
+            pl.cmap(np.ma.masked_invalid([np.nan]))[0]
+            == cmap(np.ma.masked_invalid([np.nan]))[0]
+        )
 
         # check the set_under color has been kept
-        assert pl.cmap._rgba_under == cmap._rgba_under
+        assert pl.cmap(-np.inf) == cmap(-np.inf)
 
         # check the set_over color has been kept
-        assert pl.cmap._rgba_over == cmap._rgba_over
+        assert pl.cmap(np.inf) == cmap(np.inf)
 
     def test3d(self):
         self.darray.plot()