Exemple #1
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('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()