add comparison of modifying integrator functionality to benchmarks (#35)

docs/src/benchmark_python_bseries.py

@@ -15,6 +15,10 @@
 midpoint_method = rk.loadRKM("Mid22")
 up_to_order = 9
 
+
+with open(io_file, 'a') as io:
+  print("\nModified equation", file=io)
+
 start_time = time.time()
 series = bs.modified_equation(None, None,
                               midpoint_method.A, midpoint_method.b,
@@ -34,3 +38,27 @@
 with open(io_file, 'a') as io:
   print(result, file=io)
   print("", end_time - start_time, "seconds", file=io)
+
+
+with open(io_file, 'a') as io:
+  print("\nModifying integrator", file=io)
+
+start_time = time.time()
+series = bs.modifying_integrator(None, None,
+                                 midpoint_method.A, midpoint_method.b,
+                                 up_to_order, True)
+result = sum(series.values())
+end_time = time.time()
+with open(io_file, 'a') as io:
+  print(result, file=io)
+  print("", end_time - start_time, "seconds", file=io)
+
+start_time = time.time()
+series = bs.modifying_integrator(None, None,
+                                 midpoint_method.A, midpoint_method.b,
+                                 up_to_order, True)
+result = sum(series.values())
+end_time = time.time()
+with open(io_file, 'a') as io:
+  print(result, file=io)
+  print("", end_time - start_time, "seconds", file=io)

docs/src/benchmark_python_pybs.py

@@ -20,6 +20,10 @@
 def first_values(f, n):
   return (f(tree) for tree in islice(pybs.unordered_tree.tree_generator(), 0, n))
 
+
+with open(io_file, 'a') as io:
+  print("\nModified equation", file=io)
+
 start_time = time.time()
 midpoint_series = midpoint_method.phi()
 series = pybs.series.modified_equation(midpoint_series)

docs/src/benchmarks.md

@@ -161,7 +161,8 @@ Markdown.parse(results)                                                   # hide
 
 Next, we look at the Python package
 [`pybs`](https://github.com/henriksu/pybs)
-and the following benchmark script.
+and the following benchmark script. Note that this package does not provide
+functionality for modifying integrators.
 
 ````@example
 using BSeries, Markdown                                                   # hide
@@ -201,6 +202,8 @@ b = @SArray [0, 1//1]
 c = @SArray [0, 1//2]
 up_to_order = 9
 
+
+println("Modified equation")
 @time begin
   series = modified_equation(A, b, c, up_to_order)
   println(sum(values(series)))
@@ -210,6 +213,18 @@ end
   series = modified_equation(A, b, c, up_to_order)
   println(sum(values(series)))
 end
+
+
+println("\nModifying integrator")
+@time begin
+  series = modifying_integrator(A, b, c, up_to_order)
+  println(sum(values(series)))
+end
+
+@time begin
+  series = modifying_integrator(A, b, c, up_to_order)
+  println(sum(values(series)))
+end
 ```