Merge pull request #35 from MHenderson/22-import-as

Harmonize import statements

tests/test_node_list-colouring.py → tests/test_node-list-colouring.py

@@ -1,17 +1,17 @@
-import listcolouring
+import listcolouring as lc
 import networkx as nx
-import vizing
+import vizing as vz
 
 def test_node_list_colouring_problem():
   G = nx.petersen_graph()
-  G = listcolouring.list_init_node(G, range(1, 10), 3, 0)
-  G = vizing.node_list_colouring_problem(G)
+  G = lc.list_init_node(G, range(1, 10), 3, 0)
+  G = vz.node_list_colouring_problem(G)
 
 def test_node_list_colouring_solution():
   G = nx.complete_graph(3)
   permissible_dict = {0: [0, 1], 1: [1, 2], 2: [2, 3]}
   nx.set_node_attributes(G, permissible_dict, "permissible")
-  G = vizing.node_list_colouring_solution(G)
+  G = vz.node_list_colouring_solution(G)
   assert G.nodes[0]['colour'] == 1
   assert G.nodes[1]['colour'] == 2
   assert G.nodes[2]['colour'] == 3