def test_stops_updated(self):
gradient = self.create_gradient()
self.set_stops(gradient)
with self.assertTraitChanges(gradient, "updated", count=1):
gradient.stops = [
ColorStop(offset=0.0, color='blue'),
ColorStop(offset=1.0, color='lime'),
]
def test_color_updated(self):
color_stop = ColorStop(offset=0.5, color="red")
with self.assertTraitChanges(color_stop, "updated", count=1):
color_stop.color = "blue"
def test_to_array(self):
color_stop = ColorStop(offset=0.5, color="red")
a = color_stop.to_array()
assert_array_equal(a, np.array([0.5, 1.0, 0.0, 0.0, 1.0]))
def set_brush_stops(self, brush):
brush.gradient.stops = [
ColorStop(offset=0.0, color='red'),
ColorStop(offset=0.5, color='yellow'),
ColorStop(offset=1.0, color='lime'),
]
def set_stops(self, gradient):
gradient.stops = [
ColorStop(offset=0.0, color='red'),
ColorStop(offset=0.5, color='yellow'),
ColorStop(offset=1.0, color='lime'),
]