def test_apply_cytoscape_layout_ping_failed(self):
p = MagicMock()
p.layout = 'grid'
mockpy4 = MagicMock()
mockpy4.cytoscape_ping = MagicMock(side_effect=Exception('error'))
loader = NDExNciPidLoader(p, py4cyto=mockpy4)
net = NiceCXNetwork()
try:
loader._apply_cytoscape_layout(net)
self.fail('Expected NDExNciPidLoaderError')
except NDExNciPidLoaderError as e:
self.assertEqual('Cytoscape needs to be running '
'to run layout: grid', str(e))
def test_apply_cytoscape_layout_networks_not_in_dict(self):
temp_dir = tempfile.mkdtemp()
try:
p = MagicMock()
p.layout = 'grid'
p.sifdir = temp_dir
mockpy4 = MagicMock()
mockpy4.import_network_from_file = MagicMock(return_value={})
loader = NDExNciPidLoader(p, py4cyto=mockpy4)
net = NiceCXNetwork()
for x in range(10):
net.create_node('node' + str(x))
try:
loader._apply_cytoscape_layout(net)
self.fail('Expected NDExNciPidLoaderError')
except NDExNciPidLoaderError as e:
self.assertTrue(str(e).startswith('Error network view'))
finally:
shutil.rmtree(temp_dir)
def test_apply_cytoscape_layout_networks_success(self):
temp_dir = tempfile.mkdtemp()
try:
p = MagicMock()
p.layout = 'grid'
p.sifdir = temp_dir
mockpy4 = MagicMock()
imp_res = {'networks': ['netid']}
mockpy4.import_network_from_file = MagicMock(return_value=imp_res)
mockpy4.export_network = MagicMock(return_value='')
loader = NDExNciPidLoader(p, py4cyto=mockpy4)
loader._ndexextra.extract_layout_aspect_from_cx = MagicMock(return_value={'cartesianLayout': []})
net = NiceCXNetwork()
for x in range(10):
net.create_node('node' + str(x))
loader._apply_cytoscape_layout(net)
self.assertEqual([{'cartesianLayout': []}],
net.get_opaque_aspect('cartesianLayout'))
finally:
shutil.rmtree(temp_dir)