yield self.mouse_trigger_action('add_function', pause=0) yield self.wait_for_modal() dlg = self.get_active_modal_widget() tree = get_child(dlg, 'fitTree') item = tree.findItems('Background', Qt.MatchExactly)[0] item.setExpanded(True) item = tree.findItems('BSpline', Qt.MatchRecursive)[0] pos = tree.visualItemRect(item).center() yield click_on(tree.viewport(), pos, pause=10) dlg.accept() self.assertEqual(browser.getGlobalParameters(), ['f0.A0']) @function_browser_test(multi=True) def test_converting_to_single_retains_globals(self): assert (isinstance(self, TestFunctionBrowser)) browser = self.widget browser.setFunction('name=FlatBackground;name=BSpline') browser.setGlobalParameters(['f0.A0']) view = browser.view() pos = view.getVisualRectFunctionProperty('f1.').center() tree = view.treeWidget().viewport() yield self.show_context_menu(tree, pos, pause=0) yield self.mouse_trigger_action('remove_function', pause=0) self.assertEqual(browser.getGlobalParameters(), ['A0']) if __name__ == '__main__': unittest.main() FrameworkManager.clear()
user = BrowserUser(browser) pos = view.getVisualRectFunctionProperty('f1.').center() tree = view.treeWidget().viewport() yield self.show_context_menu(tree, pos, pause=0) yield self.mouse_trigger_action('add_function', pause=0) yield self.wait_for_modal() dlg = self.get_active_modal_widget() tree = get_child(dlg, 'fitTree') item = tree.findItems('Background', Qt.MatchExactly)[0] item.setExpanded(True) item = tree.findItems('LinearBackground', Qt.MatchRecursive)[0] pos = tree.visualItemRect(item).center() yield click_on(tree.viewport(), pos) dlg.accept() fun = self.get_fit_function() self.assertEqual(fun.name, 'CompositeFunction') self.assertEqual(fun[0].name, 'FlatBackground') self.assertEqual(fun[1].name, 'CompositeFunction') self.assertEqual(fun[1][0].name, 'FlatBackground') self.assertEqual(fun[1][1].name, 'FlatBackground') self.assertEqual(fun[1][2].name, 'LinearBackground') self.assertEqual(fun[1][0].A0, 1.0) self.assertEqual(fun[1][1].A0, 2.0) self.assertEqual(fun[1][2].A0, 0.0) self.assertEqual(user.structure_changed.call_count, 1) if __name__ == '__main__': unittest.main() FrameworkManager.clear()