def createParam(self, name, value="", cpp_type="string", options=[], required=False, user_added=False, basic_type="String"): p = ParameterInfo(None, name) p.value = value p.cpp_type = cpp_type p.basic_type = basic_type p.options = options p.required = required p.user_added = user_added return p
def createParam(self, name, value="", cpp_type="string", options=[], required=False, user_added=False, group="Main"): p = ParameterInfo(None, name) p.value = value p.cpp_type = cpp_type p.options = options p.required = required p.user_added = user_added p.group_name = group return p
def testBasic(self): p = ParameterInfo(None, "p0") y = self.createData("p1", default="foo", cpp_type="some type", description="description", group_name="group", required=True) p.setFromData(y) y["default"] = "foo" self.checkParameter(p, "p1", value="foo", default="foo", cpp_type="some type", description="description", group_name="group", required=True)
def testTypes(self): b = BlockInfo(None, "/Foo", False, "") self.assertEqual(b.paramValue("type"), None) b.setBlockType("t") # shouldn't do anything p = ParameterInfo(None, "type") b.addParameter(p) self.assertEqual(p.parent, b) self.assertEqual(b.paramValue("type"), "") b.setBlockType("t") # shouldn't do anything self.assertEqual(b.paramValue("type"), "t") b1 = BlockInfo(None, "/Foo/t", False, "") p = ParameterInfo(None, "p") b1.addParameter(p) b.types[b1.name] = b1 self.assertEqual(b.paramValue("p"), "") b.setParamValue("p", "val") self.assertEqual(b.paramValue("p"), "val") b.setBlockType("t1") self.assertEqual(b.paramValue("p"), None)
def testUserParams(self): b = BlockInfo(None, "/Foo", False, "") p = ParameterInfo(None, "p") p.user_added = False b.addParameter(p) b.addUserParam("bar", "val") b.addUserParam("bar", "val") p = b.getParamInfo("bar") self.assertEqual(p.user_added, True) self.assertEqual(b.paramValue("bar"), "val") b.setParamValue("bar", "val1") self.assertEqual(b.paramValue("bar"), "val1") b.removeUserParam("bar1") self.assertEqual(b.paramValue("bar"), "val1") b.removeUserParam("p") self.assertEqual(b.paramValue("p"), "") b.removeUserParam("bar") self.assertEqual(b.paramValue("bar"), None) self.assertNotIn("bar", b.parameters_list) self.assertNotIn("bar", b.parameters) b.addUserParam("bar", "val") b.addUserParam("foo", "val1") self.assertEqual(len(b.parameters_list), 3) self.assertEqual(b.parameters_list.index("bar"), 1) self.assertEqual(b.parameters_list.index("foo"), 2) b.moveUserParam("foo", 0) self.assertEqual(b.parameters_list.index("bar"), 2) self.assertEqual(b.parameters_list.index("foo"), 0) b.renameUserParam("bar1", "bar2") b.renameUserParam("p", "bar2") self.assertEqual(b.paramValue("bar2"), None) b.renameUserParam("bar", "bar1") self.assertEqual(b.paramValue("bar"), None) self.assertEqual(b.paramValue("bar1"), "val")
def testTypes(self): p = ParameterInfo(None, "p0") y = self.createData("p1", cpp_type="vector<string>", basic_type="Array", default=None) p.setFromData(y) self.assertEqual(p.needsQuotes(), True) self.assertEqual(p.isVectorType(), True) self.assertEqual(p.default, "") p.value = "foo" self.assertEqual(p.inputFileValue(), "'foo'") y = self.createData("p1", cpp_type="bool", basic_type="Boolean", default="0") p.setFromData(y) self.assertEqual(p.value, "false") self.assertEqual(p.default, "false") self.assertEqual(p.needsQuotes(), False) self.assertEqual(p.isVectorType(), False) self.assertEqual(p.inputFileValue(), "false") y = self.createData("p1", cpp_type="bool", basic_type="Boolean", default="1") p.setFromData(y) self.assertEqual(p.value, "true") self.assertEqual(p.default, "true") y = self.createData("p1", cpp_type="bool") p.setFromData(y) self.assertEqual(p.value, "false") self.assertEqual(p.default, "false")
def testDump(self): p = ParameterInfo(None, "p0") o = StringIO() p.dump(o) val = o.getvalue() self.assertIn("Name", val)
def testCopy(self): p = ParameterInfo(None, "p0") p1 = p.copy(None) self.assertEqual(p.__dict__, p1.__dict__)
def testDump(self): p = ParameterInfo(None, "p0") o = cStringIO.StringIO() p.dump(o) val = o.getvalue() self.assertIn("Name", val)
def testAdd(self): """ Test that changing the input file resets the VTK window correctly. """ # The tabs to switch between input_ = self._app.main_widget.tab_plugin.InputFileEditorWithMesh exodus = self._app.main_widget.tab_plugin.ExodusViewer execute = self._app.main_widget.tab_plugin.ExecuteTabPlugin # The variable plugin var_plugin = exodus.currentWidget().FilePlugin blk_selector = exodus.currentWidget().BlockPlugin.BlockSelector cmap_plugin = exodus.currentWidget().ColorbarPlugin # Run and check output self.selectTab(execute) self.execute() self.selectTab(exodus) Testing.process_events(t=2) self.assertEqual([ var_plugin.VariableList.itemText(i) for i in range(var_plugin.VariableList.count()) ], ['u']) self.assertEqual([ blk_selector.StandardItemModel.item(i).text() for i in range(1, blk_selector.StandardItemModel.rowCount()) ], ['0']) # Change the colormap (to test that the colormap is maintained) idx = [ cmap_plugin.ColorMapList.itemText(i) for i in range(cmap_plugin.ColorMapList.count()) ].index('magma') cmap_plugin.ColorMapList.setCurrentIndex(idx) cmap_plugin.ColorMapList.currentIndexChanged.emit(idx) # Select the 0 block (to test that the block section is maintained) blk_selector.StandardItemModel.item(0).setCheckState(QtCore.Qt.Checked) self.assertEqual( blk_selector.StandardItemModel.item(0).checkState(), QtCore.Qt.Checked) self.assertEqual( blk_selector.StandardItemModel.item(1).checkState(), QtCore.Qt.Checked) # Add a variable self.selectTab(input_) b0 = input_.InputFileEditorPlugin.tree.getBlockInfo("/AuxVariables") b0.included = True input_.InputFileEditorPlugin.block_tree.copyBlock(b0) input_.InputFileEditorPlugin.blockChanged.emit( b0, input_.InputFileEditorPlugin.tree) # Add MeshModifier b1 = input_.InputFileEditorPlugin.tree.getBlockInfo("/MeshModifiers") b1.included = True input_.InputFileEditorPlugin.block_tree.copyBlock(b1) b2 = list(b1.children.values())[0] b2.setBlockType('SubdomainBoundingBox') b2.addParameter(ParameterInfo(None, 'bottom_left')) b2.setParamValue('bottom_left', '0.25 0.25 0') b2.addParameter(ParameterInfo(None, 'top_right')) b2.setParamValue('top_right', '0.75 0.75 0') b2.addParameter(ParameterInfo(None, 'block_id')) b2.setParamValue('block_id', '1980') input_.InputFileEditorPlugin.blockChanged.emit( b1, input_.InputFileEditorPlugin.tree) # Run and check output self.selectTab(execute) Testing.process_events(t=2) self.execute() self.selectTab(exodus) self.assertEqual([ var_plugin.VariableList.itemText(i) for i in range(var_plugin.VariableList.count()) ], ['New_0', 'u']) # Check colormap self.assertEqual(self._window._result.getOption('cmap'), 'default') self.assertEqual(cmap_plugin.ColorMapList.currentText(), 'default') # Check variable self.assertEqual(self._window._result.getOption('variable'), 'New_0') self.assertEqual(var_plugin.VariableList.currentText(), 'New_0') # Check blocks self.assertEqual([ blk_selector.StandardItemModel.item(i).text() for i in range(1, blk_selector.StandardItemModel.rowCount()) ], ['0', '1980']) self.assertEqual( blk_selector.StandardItemModel.item(0).checkState(), QtCore.Qt.Checked) self.assertEqual( blk_selector.StandardItemModel.item(1).checkState(), QtCore.Qt.Checked) self.assertEqual( blk_selector.StandardItemModel.item(2).checkState(), QtCore.Qt.Checked)