def METHOD_NAME(self):
    self.colormap.color_depth = "rgb"
    a = array([0, 2, 3])
    b = self.colormap.map_screen(a)
    self.assertEqual(b.shape, (3, 3))
    for i in range(3):
        assert_array_almost_equal(b[:, i], array([0.0, 0.5, 0.75]))