class TreeSetUp(unittest.TestCase):
def setUp(self):
self.createRootNode()
self.createNodesLevel_1()
self.createBranch_1_FEM()
self.createBranch_1_XFEM()
self.createBranch_2_FEM()
self.createBranch_3_FEM()
self.createBranch_3_XFEM()
self.createBranch_5_XFEM()
self.createBranch_10_FEM()
self.createBranch_10_XFEM()
self.add_simIds_to_tree_leaf_nodes()
def createRootNode(self):
self.root = TreeNode('root')
def createNodesLevel_1(self):
self.node1 = TreeNode('1.0')
self.root.setChild(self.node1)
self.node1.setParent(self.root)
self.node2 = TreeNode('2.0')
self.root.setChild(self.node2)
self.node2.setParent(self.root)
self.node3 = TreeNode('3.0')
self.root.setChild(self.node3)
self.node3.setParent(self.root)
self.node5 = TreeNode('5.0')
self.root.setChild(self.node5)
self.node5.setParent(self.root)
self.node10 = TreeNode('10.0')
self.root.setChild(self.node10)
self.node10.setParent(self.root)
def createBranch_1_FEM(self):
self.node1FEM = TreeNode('FEM')
self.node1.setChild(self.node1FEM)
self.node1FEM.setParent(self.node1)
self.node1FEMs = TreeNode('scale')
self.node1FEM.setChild(self.node1FEMs)
self.node1FEMs.setParent(self.node1FEM)
self.node1FEMsQR = TreeNode('QR')
self.node1FEMs.setChild(self.node1FEMsQR)
self.node1FEMsQR.setParent(self.node1FEMs)
self.node1FEMsLR = TreeNode('LR')
self.node1FEMs.setChild(self.node1FEMsLR)
self.node1FEMsLR.setParent(self.node1FEMs)
self.node1FEMe = TreeNode('elliptic')
self.node1FEM.setChild(self.node1FEMe)
self.node1FEMe.setParent(self.node1FEM)
self.node1FEMeQF = TreeNode('QF')
self.node1FEMe.setChild(self.node1FEMeQF)
self.node1FEMeQF.setParent(self.node1FEMe)
self.node1FEMeLF = TreeNode('LF')
self.node1FEMe.setChild(self.node1FEMeLF)
self.node1FEMeLF.setParent(self.node1FEMe)
def createBranch_1_XFEM(self):
self.node1XFEM = TreeNode('XFEM')
self.node1.setChild(self.node1XFEM)
self.node1XFEM.setParent(self.node1)
self.node1XFEMcp = TreeNode('cp')
self.node1XFEM.setChild(self.node1XFEMcp)
self.node1XFEMcp.setParent(self.node1XFEM)
self.node1XFEMcpLT = TreeNode('LT')
self.node1XFEMcp.setChild(self.node1XFEMcpLT)
self.node1XFEMcpLT.setParent(self.node1XFEMcp)
self.node1XFEMmp = TreeNode('mp')
self.node1XFEMmp.setParent(self.node1XFEM)
self.node1XFEM.setChild(self.node1XFEMmp)
self.node1XFEMmpLR = TreeNode('LR')
self.node1XFEMmp.setChild(self.node1XFEMmpLR)
self.node1XFEMmpLR.setParent(self.node1XFEMmp)
def createBranch_2_FEM(self):
self.node2FEM = TreeNode('FEM')
self.node2.setChild(self.node2FEM)
self.node2FEM.setParent(self.node2)
self.node2FEMs = TreeNode('scale')
self.node2FEM.setChild(self.node2FEMs)
self.node2FEMs.setParent(self.node2FEM)
self.node2FEMsQF = TreeNode('QF')
self.node2FEMsQF.setParent(self.node2FEMs)
self.node2FEMs.setChild(self.node2FEMsQF)
def createBranch_3_FEM(self):
self.node3FEM = TreeNode('FEM')
self.node3.setChild(self.node3FEM)
self.node3FEM.setParent(self.node3)
self.node3FEMe = TreeNode('elliptic')
self.node3FEM.setChild(self.node3FEMe)
self.node3FEMe.setParent(self.node3FEM)
self.node3FEMeQR = TreeNode('QR')
self.node3FEMe.setChild(self.node3FEMeQR)
self.node3FEMeQR.setParent(self.node3FEMe)
def createBranch_3_XFEM(self):
self.node3XFEM = TreeNode('XFEM')
self.node3.setChild(self.node3XFEM)
self.node3XFEM.setParent(self.node3)
self.node3XFEMmp = TreeNode('mp')
self.node3XFEM.setChild(self.node3XFEMmp)
self.node3XFEMmp.setParent(self.node3XFEM)
self.node3XFEMmpLF = TreeNode('LF')
self.node3XFEMmp.setChild(self.node3XFEMmpLF)
self.node3XFEMmpLF.setParent(self.node3XFEMmp)
def createBranch_5_XFEM(self):
self.node5XFEM = TreeNode('XFEM')
self.node5XFEM.setParent(self.node5)
self.node5.setChild(self.node5XFEM)
self.node5XFEMcp = TreeNode('cp')
self.node5XFEMcp.setParent(self.node5XFEM)
self.node5XFEM.setChild(self.node5XFEMcp)
self.node5XFEMcpLT = TreeNode('LT')
self.node5XFEMcp.setChild(self.node5XFEMcpLT)
self.node5XFEMcpLT.setParent(self.node5XFEMcp)
def createBranch_10_FEM(self):
self.node10FEM = TreeNode('FEM')
self.node10.setChild(self.node10FEM)
self.node10FEM.setParent(self.node10)
self.node10FEMe = TreeNode('elliptic')
self.node10FEMe.setParent(self.node10FEM)
self.node10FEM.setChild(self.node10FEMe)
self.node10FEMeQF = TreeNode('QF')
self.node10FEMe.setChild(self.node10FEMeQF)
self.node10FEMeQF.setParent(self.node10FEMe)
self.node10FEMeLF = TreeNode('LF')
self.node10FEMe.setChild(self.node10FEMeLF)
self.node10FEMeLF.setParent(self.node10FEMe)
def createBranch_10_XFEM(self):
self.node10XFEM = TreeNode('XFEM')
self.node10.setChild(self.node10XFEM)
self.node10XFEM.setParent(self.node10)
self.node10XFEMcp = TreeNode('cp')
self.node10XFEM.setChild(self.node10XFEMcp)
self.node10XFEMcp.setParent(self.node10XFEM)
self.node10XFEMcpLR = TreeNode('LR')
self.node10XFEMcp.setChild(self.node10XFEMcpLR)
self.node10XFEMcpLR.setParent(self.node10XFEMcp)
def add_simIds_to_tree_leaf_nodes(self):
# branch 1.0
self.node1FEMsQR.successfulMembers = set(
['node1FEMsQR_sm_1', 'node1FEMsQR_sm_2'])
self.node1FEMsQR.failedMembers = set(['node1FEMsQR_fm_1'])
self.node1FEMsLR.successfulMembers = set(['node1FEMsLR_sm_1'])
self.node1FEMsLR.failedMembers = set(['node1FEMsLR_fm_1'])
self.node1FEMeQF.successfulMembers = set(['node1FEMeQF_sm_1'])
self.node1FEMeQF.failedMembers = set(
['node1FEMeQF_fm_1', 'node1FEMeQF_fm_2'])
self.node1FEMeLF.successfulMembers = set(
['node1FEMeLF_sm_1', 'node1FEMeLF_sm_2', 'node1FEMeLF_sm_3'])
self.node1FEMeLF.failedMembers = set(['node1FEMeLF_fm_1'])
self.node1XFEMcpLT.successfulMembers = set(
['node1XFEMcpLT_sm_1', 'node1XFEMcpLT_sm_2', 'node1XFEMcpLT_sm_3'])
self.node1XFEMcpLT.failedMembers = set(['node1XFEMcpLT_fm_1'])
self.node1XFEMmpLR.successfulMembers = set(
['node1XFEMmpLR_sm_1', 'node1XFEMmpLR_sm_2'])
self.node1XFEMmpLR.failedMembers = set(['node1XFEMmpLR_fm_1'])
# branch 2.0
self.node2FEMsQF.successfulMembers = set(
['node2FEMsQF_sm_1', 'node2FEMsQF_sm_2'])
# self.node2FEMsQF.failedMembers # None
# branch 3.0
self.node3FEMeQR.successfulMembers = set(['node3FEMeQR_sm_1'])
self.node3FEMeQR.failedMembers = set(
['node3FEMeQR_fm_1', 'node3FEMeQR_fm_2'])
self.node3XFEMmpLF.successfulMembers = set(
['node3XFEMmpLF_sm_1', 'node3XFEMmpLF_sm_2'])
self.node3XFEMmpLF.failedMembers = set(
['node3XFEMmpLF_fm_1', 'node3XFEMmpLF_fm_2', 'node3XFEMmpLF_fm_3'])
# branch 5.0
self.node5XFEMcpLT.successfulMembers = set(
['node5XFEMcpLT_sm_1', 'node5XFEMcpLT_sm_2'])
self.node5XFEMcpLT.failedMembers = set(
['node5XFEMcpLT_fm_1', 'node5XFEMcpLT_fm_2', 'node5XFEMcpLT_fm_3'])
# branch 10.0
self.node10FEMeQF.successfulMembers = set(
['node10FEMeQF_sm_1', 'node10FEMeQF_sm_2'])
self.node10FEMeQF.failedMembers = set(
['node10FEMeQF_fm_1', 'node10FEMeQF_fm_2'])
self.node10FEMeLF.successfulMembers = set(
['node10FEMeLF_sm_1', 'node10FEMeLF_sm_2', 'node10FEMeLF_sm_3'])
self.node10FEMeLF.failedMembers = set(
['node10FEMeLF_fm_1', 'node10FEMeLF_fm_2', 'node10FEMeLF_fm_3'])
# self.node10XFEMcpLR.successfulMembers = set([])
self.node10XFEMcpLR.failedMembers = set(
['node10XFEMcpLR_fm_1', 'node10XFEMcpLR_fm_2'])
