minor fixes in sandpiles

Author: fchapoton

src/sage/sandpiles/examples.py

@@ -25,7 +25,8 @@
 from sage.sandpiles.sandpile import Sandpile
 from sage.graphs.graph_generators import graphs
 
-class SandpileExamples():
+
+class SandpileExamples:
     """
     Some examples of sandpiles.
 
@@ -92,7 +93,7 @@ def Complete(self, n):
             sage: sandpiles.Complete(3) == sandpiles.Cycle(3)
             True
         """
-        return Sandpile(graphs.CompleteGraph(n),0)
+        return Sandpile(graphs.CompleteGraph(n), 0)
 
     def Cycle(self, n):
         """
@@ -119,7 +120,7 @@ def Cycle(self, n):
              (3, 0, 1),
              (3, 2, 1)]
         """
-        return Sandpile(graphs.CycleGraph(n),0)
+        return Sandpile(graphs.CycleGraph(n), 0)
 
     def Diamond(self):
         """
@@ -164,18 +165,18 @@ def Fan(self, n, deg_three_verts=False):
         """
         f = graphs.WheelGraph(n)
         if n > 2:
-            f.delete_edge(1,n-1)
+            f.delete_edge(1, n-1)
             if deg_three_verts:
                 f.allow_multiple_edges(True)
-                f.add_edges([(0,1),(0,n-1)])
-            return Sandpile(f,0)
+                f.add_edges([(0, 1), (0, n-1)])
+            return Sandpile(f, 0)
         elif n == 1:
-            return Sandpile(f,0)
+            return Sandpile(f, 0)
         elif n == 2:
             if deg_three_verts:
-                return Sandpile({0:{1:3}, 1:{0:3}})
+                return Sandpile({0: {1: 3}, 1: {0: 3}})
             else:
-                return Sandpile(f,0)
+                return Sandpile(f, 0)
 
     def Grid(self, m, n):
         """
@@ -200,11 +201,12 @@ def Grid(self, m, n):
             sage: s.dict()
             {(0, 0): {(1, 1): 4}, (1, 1): {(0, 0): 4}}
         """
-        G = graphs.Grid2dGraph(m+2,n+2)
+        G = graphs.Grid2dGraph(m+2, n+2)
         G.allow_multiple_edges(True)  # to ensure each vertex ends up with degree 4
-        V = [(i,j) for i in [0,m+1] for j in range(n+2)] + [(i,j) for j in [0,n+1] for i in range(m+2)]
+        V = [(i, j) for i in [0, m+1] for j in range(n+2)]
+        V += [(i, j) for j in [0, n+1] for i in range(m+2)]
         G.merge_vertices(V)
-        return Sandpile(G, (0,0))
+        return Sandpile(G, (0, 0))
 
     def House(self):
         """
@@ -224,7 +226,7 @@ def House(self):
             sage: s.invariant_factors()
             [1, 1, 1, 11]
         """
-        return Sandpile(graphs.HouseGraph(),0)
+        return Sandpile(graphs.HouseGraph(), 0)
 
     def Wheel(self, n):
         """
@@ -244,7 +246,7 @@ def Wheel(self, n):
             sage: w.invariant_factors()
             [1, 1, 1, 11, 11]
         """
-        return Sandpile(graphs.WheelGraph(n),0)
+        return Sandpile(graphs.WheelGraph(n), 0)
 
 
 sandpiles = SandpileExamples()