From 5c95115d7c282bf5f605a132a95300b27dd76dcf Mon Sep 17 00:00:00 2001 From: Matthew Horton Date: Wed, 6 Nov 2019 09:37:31 -0800 Subject: [PATCH] Update syntax for new networkx --- pymatgen/analysis/tests/test_graphs.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/pymatgen/analysis/tests/test_graphs.py b/pymatgen/analysis/tests/test_graphs.py index 8b6f7cababa..be43323d6a1 100644 --- a/pymatgen/analysis/tests/test_graphs.py +++ b/pymatgen/analysis/tests/test_graphs.py @@ -694,13 +694,13 @@ def test_construction(self): self.assertEqual(mol_graph.graph.adj, ref_mol_graph.graph.adj) for node in mol_graph.graph.nodes: self.assertEqual( - mol_graph.graph.node[node]["specie"], - ref_mol_graph.graph.node[node]["specie"], + mol_graph.graph.nodes[node]["specie"], + ref_mol_graph.graph.nodes[node]["specie"], ) for ii in range(3): self.assertEqual( - mol_graph.graph.node[node]["coords"][ii], - ref_mol_graph.graph.node[node]["coords"][ii], + mol_graph.graph.nodes[node]["coords"][ii], + ref_mol_graph.graph.nodes[node]["coords"][ii], ) edges_pc = {(e[0], e[1]): {"weight": 1.0} for e in self.pc_edges} @@ -711,13 +711,13 @@ def test_construction(self): self.assertEqual(mol_graph.graph.adj, ref_mol_graph.graph.adj) for node in mol_graph.graph: self.assertEqual( - mol_graph.graph.node[node]["specie"], - ref_mol_graph.graph.node[node]["specie"], + mol_graph.graph.nodes[node]["specie"], + ref_mol_graph.graph.nodes[node]["specie"], ) for ii in range(3): self.assertEqual( - mol_graph.graph.node[node]["coords"][ii], - ref_mol_graph.graph.node[node]["coords"][ii], + mol_graph.graph.nodes[node]["coords"][ii], + ref_mol_graph.graph.nodes[node]["coords"][ii], ) mol_graph_edges = MoleculeGraph.with_edges(self.pc, edges=edges_pc)