Fix tests

Co-Authored-By: Daniel Domingo-Fernández <[email protected]>
bio2bel · Apr 18, 2018 · a828563 · a828563
1 parent a437bd0
commit a828563
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_enrich.py b/tests/test_enrich.py
@@ -16,15 +16,15 @@ def test_get_pathway_graph(self):
     def test_enrich_wikipathway_pathway(self):
         graph_example = enrichment_graph()
 
-        enriched_graph = self.manager.enrich_wikipathways_pathway(graph_example)
+        self.manager.enrich_wikipathways_pathway(graph_example)
 
-        self.assertEqual(6, enriched_graph.number_of_nodes())  # 4 nodes + 2 new
-        self.assertEqual(5, enriched_graph.number_of_edges())  # 3 + 2 new
+        self.assertEqual(6, graph_example.number_of_nodes())  # 4 nodes + 2 new
+        self.assertEqual(5, graph_example.number_of_edges())  # 3 + 2 new
 
     def test_enrich_wikipathway_protein(self):
         graph_example = enrichment_graph()
 
-        enriched_graph = self.manager.enrich_wikipathways_protein(graph_example)
+        self.manager.enrich_wikipathways_protein(graph_example)
 
-        self.assertEqual(6, enriched_graph.number_of_nodes())  # 4 +  2 new pathways
-        self.assertEqual(5, enriched_graph.number_of_edges())  # 3 + 2 new
+        self.assertEqual(6, graph_example.number_of_nodes())  # 4 +  2 new pathways
+        self.assertEqual(5, graph_example.number_of_edges())  # 3 + 2 new