Merge pull request #2 from isosafrasaurus/main

Fix demo code in plot.py and not calling super().__init__()

Author: MiroK

graphnics/plot.py

@@ -40,15 +40,15 @@ def __init__(self, G, fname, **kwargs):
             fname (str): location and name for file
             
         Example:
-        >> G = make_Y_bifurcation(dim=3)
+        >> G = Y_bifurcation(dim=3)
         >> for e in G.edges():
-        >>     G.edges()[e]['radius'] = 1
-        >> V = FunctionSpace(G.mesh, 'CG', 1)
-        >> f_i = interpolate(Expression('x[0]', degree=2))
-        >> f_i.rename('f', '0.0')
-        >> TubeFile(G, 'test.pvd') << f_i  
+        >>      G.edges()[e]['radius'] = 1
+        >>  V = FunctionSpace(G.mesh, 'CG', 1)
+        >>  f_i = interpolate(Expression('x[0]', degree=2), V)
+        >>  f_i.rename('f', '0.0')
+        >>  TubeFile(G, 'test.pvd') << f_i  
         """
-        
+        super().__init__(fname)
         f_name, f_ext = os.path.splitext(fname)
         assert f_ext == '.pvd', 'TubeFile must have .pvd file ending'
 
@@ -167,4 +167,4 @@ def __lshift__(self, func):
   <Collection>\n"""
 
 pvd_footer= """</Collection>
-</VTKFile>"""
+</VTKFile>"""