Python _build_discrete_cmap Examples

Example #1

File: test_plot.py Project: spencerahill/xarray

 def test_build_discrete_cmap(self):
     for (cmap, levels, extend, filled) in [("jet", [0, 1], "both", False), ("hot", [-4, 4], "max", True)]:
         ncmap, cnorm = _build_discrete_cmap(cmap, levels, extend, filled)
         self.assertEqual(ncmap.N, len(levels) - 1)
         self.assertEqual(len(ncmap.colors), len(levels) - 1)
         self.assertEqual(cnorm.N, len(levels))
         self.assertArrayEqual(cnorm.boundaries, levels)
         self.assertEqual(max(levels), cnorm.vmax)
         self.assertEqual(min(levels), cnorm.vmin)
         if filled:
             self.assertEqual(ncmap.colorbar_extend, extend)
         else:
             self.assertEqual(ncmap.colorbar_extend, "neither")

Example #2

 def test_build_discrete_cmap(self):
     for (cmap, levels, extend, filled) in [('jet', [0, 1], 'both', False),
                                            ('hot', [-4, 4], 'max', True)]:
         ncmap, cnorm = _build_discrete_cmap(cmap, levels, extend, filled)
         self.assertEqual(ncmap.N, len(levels) - 1)
         self.assertEqual(len(ncmap.colors), len(levels) - 1)
         self.assertEqual(cnorm.N, len(levels))
         self.assertArrayEqual(cnorm.boundaries, levels)
         self.assertEqual(max(levels), cnorm.vmax)
         self.assertEqual(min(levels), cnorm.vmin)
         if filled:
             self.assertEqual(ncmap.colorbar_extend, extend)
         else:
             self.assertEqual(ncmap.colorbar_extend, 'neither')

Example #3

File: test_plot.py Project: CCI-Tools/xarray

 def test_build_discrete_cmap(self):
     for (cmap, levels, extend, filled) in [('jet', [0, 1], 'both', False),
                                            ('hot', [-4, 4], 'max', True)]:
         ncmap, cnorm = _build_discrete_cmap(cmap, levels, extend, filled)
         self.assertEqual(ncmap.N, len(levels) - 1)
         self.assertEqual(len(ncmap.colors), len(levels) - 1)
         self.assertEqual(cnorm.N, len(levels))
         self.assertArrayEqual(cnorm.boundaries, levels)
         self.assertEqual(max(levels), cnorm.vmax)
         self.assertEqual(min(levels), cnorm.vmin)
         if filled:
             self.assertEqual(ncmap.colorbar_extend, extend)
         else:
             self.assertEqual(ncmap.colorbar_extend, 'max')

Example #4

File: test_plot.py Project: thadncs/https-github.com-pydata-xarray

 def test_build_discrete_cmap(self):
     for (cmap, levels, extend, filled) in [('jet', [0, 1], 'both', False),
                                            ('hot', [-4, 4], 'max', True)]:
         ncmap, cnorm = _build_discrete_cmap(cmap, levels, extend, filled)
         assert ncmap.N == len(levels) - 1
         assert len(ncmap.colors) == len(levels) - 1
         assert cnorm.N == len(levels)
         assert_array_equal(cnorm.boundaries, levels)
         assert max(levels) == cnorm.vmax
         assert min(levels) == cnorm.vmin
         if filled:
             assert ncmap.colorbar_extend == extend
         else:
             assert ncmap.colorbar_extend == 'max'