def test_06_MSMS(): # test the MSMS node net = ed.currentNetwork net.runOnNewData.value = True node1 = ReadMolecule(library=molkitlib) net.addNode(node1, 20, 20) node1.inputPorts[0].widget.set(os.path.abspath("1crn.pdb")) # assign radii node2 = AssignRadii(library=molkitlib) net.addNode(node2, 275, 62) net.connectNodes(node1, node2, "MolSets", "molecules") node2.toggleNodeExpand_cb() # select nodes node3 = NodeSelector(library=molkitlib) net.addNode(node3, 14, 153) net.connectNodes(node2, node3, "molecules", "nodes") node3.toggleNodeExpand_cb() # msms node4 = AtomsAsMSMS(library=molkitlib) net.addNode(node4, 263, 267) net.connectNodes(node3, node4, "nodes", "atoms") net.connectNodes(node2, node4, "radii", "radii") pause() # check data from mslib import MSMS assert isinstance(node4.outputPorts[0].data, MSMS),\ "Expected %s, got %s"%(MSMS, node4.outputPorts[0].data.__class__) assert len(node4.outputPorts[0].data.coords) == 327,\ "Expected 327, got %s"%len(node4.outputPorts[0].data.coords) # just for the fun of it, click the united radii checkbutton a couple times node2.inputPorts[1].widget.set(1) node2.inputPorts[1].widget.set(0) node2.inputPorts[1].widget.set(1) node2.inputPorts[1].widget.set(0)
def test_02_NodeSelector(): # test the select 'node' node net = ed.currentNetwork net.runOnNewData.value = True node1 = ReadMolecule(library=molkitlib) net.addNode(node1, 20, 20) node1.inputPorts[0].widget.set(os.path.abspath("1crn.pdb")) node2 = NodeSelector(library=molkitlib) net.addNode(node2, 30, 100) net.connectNodes(node1, node2, "MolSets", "nodes") node2.toggleNodeExpand_cb() pause() # because we run(), upon connecting we pass the data and the output of # node2 should have data # default node output is AtomSet from MolKit.molecule import AtomSet data = node2.outputPorts[0].data assert isinstance( data, AtomSet), "Expected %s, got %s" % (AtomSet, data.__class__) # switch to ResidueSet from MolKit.protein import ResidueSet node2.inputPorts[1].widget.set("Residue") data = node2.outputPorts[0].data assert isinstance( data, ResidueSet), "Expected %s, got %s" % (ResidueSet, data.__class__) # switch to Chain from MolKit.protein import ChainSet node2.inputPorts[1].widget.set("Chain") data = node2.outputPorts[0].data assert isinstance( data, ChainSet), "Expected %s, got %s" % (ChainSet, data.__class__) # switch to Molecule from MolKit.molecule import MoleculeSet node2.inputPorts[1].widget.set("Molecule") data = node2.outputPorts[0].data assert isinstance( data, MoleculeSet), "Expected %s, got %s" % (MoleculeSet, data.__class__)
def test_02_NodeSelector(): # test the select 'node' node net = ed.currentNetwork net.runOnNewData.value = True node1 = ReadMolecule(library=molkitlib) net.addNode(node1, 20, 20) node1.inputPorts[0].widget.set(os.path.abspath("1crn.pdb")) node2 = NodeSelector(library=molkitlib) net.addNode(node2, 30, 100) net.connectNodes(node1, node2, "MolSets", "nodes") node2.toggleNodeExpand_cb() pause() # because we run(), upon connecting we pass the data and the output of # node2 should have data # default node output is AtomSet from MolKit.molecule import AtomSet data = node2.outputPorts[0].data assert isinstance(data, AtomSet), "Expected %s, got %s"%( AtomSet, data.__class__) # switch to ResidueSet from MolKit.protein import ResidueSet node2.inputPorts[1].widget.set("Residue") data = node2.outputPorts[0].data assert isinstance(data, ResidueSet), "Expected %s, got %s"%( ResidueSet, data.__class__) # switch to Chain from MolKit.protein import ChainSet node2.inputPorts[1].widget.set("Chain") data = node2.outputPorts[0].data assert isinstance(data, ChainSet), "Expected %s, got %s"%( ChainSet, data.__class__) # switch to Molecule from MolKit.molecule import MoleculeSet node2.inputPorts[1].widget.set("Molecule") data = node2.outputPorts[0].data assert isinstance(data, MoleculeSet), "Expected %s, got %s"%( MoleculeSet, data.__class__)