fix(stressgen): reformat with Black

1 files changed, 20 insertions, 16 deletions

diff --git a/stressgen b/stressgen
index 4fc5c0d..eb87578 100755
--- a/stressgen
+++ b/stressgen
@@ -4,33 +4,37 @@ import sys
 
 
 def generate_prelude(f, basename):
-    f.write('''\
+    f.write(
+        """\
 ---
 title: "stress test"
 bindings: {basename}.yaml
 functions: {basename}.py
 ...
-'''.format(basename=basename))
+""".format(
+            basename=basename
+        )
+    )
 
 
 def generate_scenario(f, scen_number, steps):
-    f.write('\n')
-    f.write('# Scenario {}\n'.format(scen_number))
-    f.write('~~~scenario\n')
+    f.write("\n")
+    f.write("# Scenario {}\n".format(scen_number))
+    f.write("~~~scenario\n")
     for i in range(steps):
-        f.write('given precondition {}\n'.format(i))
-    f.write('~~~\n')
-    
+        f.write("given precondition {}\n".format(i))
+    f.write("~~~\n")
+
 
 def generate_bindings(f, num_bindings):
     for i in range(num_bindings):
-        f.write('- given: precondition {}\n'.format(i))
-        f.write('  function: nop\n')
-    
+        f.write("- given: precondition {}\n".format(i))
+        f.write("  function: nop\n")
+
 
 def generate_functions(f):
-    f.write('def nop(ctx):\n')
-    f.write('  pass\n')
+    f.write("def nop(ctx):\n")
+    f.write("  pass\n")
 
 
 def main():
@@ -38,15 +42,15 @@ def main():
     scenarios = int(sys.argv[2])
     steps = int(sys.argv[3])
 
-    with open(basename + '.md', 'w') as f:
+    with open(basename + ".md", "w") as f:
         generate_prelude(f, basename)
         for i in range(scenarios):
             generate_scenario(f, i, steps)
 
-    with open(basename + '.yaml', 'w') as f:
+    with open(basename + ".yaml", "w") as f:
         generate_bindings(f, steps)
 
-    with open(basename + '.py', 'w') as f:
+    with open(basename + ".py", "w") as f:
         generate_functions(f)