def test_base(self, qtbot):
dkt = ColormapDict({})
box = ColorComboBox(0, starting_colors, dkt)
box.show()
qtbot.add_widget(box)
with qtbot.waitSignal(
box.channel_visible_changed), qtbot.assertNotEmitted(
box.clicked):
qtbot.mouseClick(box.check_box, Qt.LeftButton)
with qtbot.waitSignal(box.clicked, timeout=1000):
qtbot.mouseClick(box, Qt.LeftButton, pos=QPoint(5, 5))
with qtbot.waitSignal(box.clicked):
qtbot.mouseClick(box,
Qt.LeftButton,
pos=QPoint(box.width() - 5, 5))
index = 3
with qtbot.waitSignal(box.currentTextChanged):
box.set_color(starting_colors[index])
img = np.array(
make_colorbar(dkt[starting_colors[index]][0], size=(1, 512)))
print(array_from_image(box.image), array_from_image(box.image).size)
print(img)
print(img.flatten(), img.size, img.shape)
print(dkt[starting_colors[index]][0])
print(box.image, box.image.size(), box.image.depth())
assert np.all(array_from_image(box.image) == img.flatten())
box.hide()
def test_change_colors(self, qtbot):
dkt = ColormapDict({})
box = ColorComboBox(0, starting_colors, dkt)
qtbot.add_widget(box)
box.change_colors(starting_colors[:-1])
assert box.count() == len(starting_colors) - 1
box.change_colors(starting_colors[1:])
assert box.count() == len(starting_colors) - 1
def test_edit_button(self, qtbot):
dkt = ColormapDict({})
color_list = ColormapList(dkt, starting_colors)
color_list.refresh()
qtbot.addWidget(color_list)
color_edit = ColormapCreator()
color_list.edit_signal.connect(color_edit.set_colormap)
qtbot.addWidget(color_edit)
cmap = dkt["BlackRed"][0]
assert cmap != color_edit.current_colormap()
color_list.get_colormap_widget("BlackRed").edit_btn.click()
assert cmap == color_edit.current_colormap()
assert color_list._widget_dict["magma"].edit_btn.isEnabled() is False
def test_show_frame_arrow(self, qtbot):
dkt = ColormapDict({})
box = ColorComboBox(0, starting_colors, dkt)
qtbot.add_widget(box)
box.show()
box.show_arrow = True
box.repaint()
qtbot.wait(100)
box.show_arrow = False
box.show_frame = True
box.repaint()
qtbot.wait(100)
box.hide()
def test_visibility(self, qtbot):
dkt = ColormapDict({})
box = ColorComboBox(0, starting_colors, dkt, lock=True)
box.show()
qtbot.add_widget(box)
assert box.lock.isVisible()
box = ColorComboBox(0, starting_colors, dkt, blur=NoiseFilterType.Gauss)
box.show()
qtbot.add_widget(box)
assert box.blur.isVisible()
box = ColorComboBox(0, starting_colors, dkt, gamma=2)
box.show()
qtbot.add_widget(box)
assert box.gamma.isVisible()
def test_color_combo_box(qtbot):
dkt = ColormapDict({})
box = ColorComboBox(0, starting_colors, dkt)
box.show()
qtbot.add_widget(box)
with qtbot.waitSignal(box.channel_visible_changed):
with qtbot.assertNotEmitted(box.clicked):
qtbot.mouseClick(box.check_box, Qt.LeftButton)
with qtbot.waitSignal(box.clicked, timeout=1000):
qtbot.mouseClick(box, Qt.LeftButton, pos=QPoint(5, 5))
with qtbot.waitSignal(box.clicked):
qtbot.mouseClick(box, Qt.LeftButton, pos=QPoint(box.width() - 5, 5))
index = 3
with qtbot.waitSignal(box.currentTextChanged):
box.set_color(starting_colors[index])
img = color_bar_fun(np.linspace(0, 256, 512, endpoint=False).reshape((1, 512, 1)), dkt[starting_colors[index]][0])
assert np.all(array_from_image(box.image) == img.flatten())
def test_base(self, qtbot):
dkt = ColormapDict({})
widget = ColormapList(dkt, starting_colors)
widget.refresh()
qtbot.addWidget(widget)
with qtbot.waitSignal(widget.visibility_colormap_change,
check_params_cb=self.verify_visibility(
starting_colors[0], False)):
widget.set_state(starting_colors[0], False)
assert len(widget.get_selected()) == len(starting_colors) - 1
with qtbot.waitSignal(widget.visibility_colormap_change,
check_params_cb=self.verify_visibility(
starting_colors[0], True)):
widget.set_state(starting_colors[0], True)
assert len(widget.get_selected()) == len(starting_colors)
name = starting_colors[0] + "_reversed"
with qtbot.waitSignal(widget.visibility_colormap_change,
check_params_cb=self.verify_visibility(
name, True)):
widget.set_state(name, True)
assert len(widget.get_selected()) == len(starting_colors) + 1