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