Fix unit tests, v3

LevN0 · LevN0 · commit 8899bae9264a · 2019-01-23T00:35:24.000-05:00
Fix unit tests, v3
diff --git a/lib/matplotlib/tests/test_colorbar.py b/lib/matplotlib/tests/test_colorbar.py
@@ -311,19 +311,19 @@ def test_colorbar_minorticks_on_off():
     data = np.random.uniform(low=1, high=10, size=(20, 20))
 
     fig, ax = plt.subplots()
-    im = ax.pcolormesh(data, norm=mpl.colors.LogNorm())
+    im = ax.pcolormesh(data, norm=LogNorm())
     cbar = fig.colorbar(im)
     default_minorticklocks = cbar.ax.yaxis.get_minorticklocs()
 
     # test that minorticks turn off for LogNorm
     cbar.minorticks_off()
-    np.testing.assert_almost_equal(cbar.ax.yaxis.get_minorticklocs(),
-                                   np.array([]))
+    assert np.array_equal(cbar.ax.yaxis.get_minorticklocs(),
+                          np.array([]))
 
     # test that minorticks turn back on for LogNorm
     cbar.minorticks_on()
-    np.testing.assert_almost_equal(cbar.ax.yaxis.get_minorticklocs(),
-                                   default_minorticklocks)
+    assert np.array_equal(cbar.ax.yaxis.get_minorticklocs(),
+                          default_minorticklocks)
 
 
 def test_colorbar_autoticks():
