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 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 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 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 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 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 = cStringIO.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 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 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)
