def METHOD_NAME(self):
    a = ArrayDataSource(array([0.0, 0.5, 1.0]))
    self.colormap.range.add(a)
    b = self.colormap.map_screen(a.get_data())
    self.colormap.range.remove(a)
    expected = array([0.0, 0.5, 1.0])
    close = allclose(ravel(b[:, :1]), expected, atol=0.02)
    self.assertTrue(
        close,
        "Simple map failed.  Expected %s.  Got %s" % (expected, b[:, :1]),
    )