167 lines
5.0 KiB
Python
167 lines
5.0 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
from glob import glob
|
|
from pathlib import Path
|
|
from subprocess import PIPE, STDOUT, run
|
|
|
|
from pytablewriter import MarkdownTableWriter
|
|
from pytablewriter.style import Style
|
|
|
|
|
|
def read_md():
|
|
"""
|
|
Read README.md
|
|
"""
|
|
with open("../README.md", "r") as f:
|
|
md = [line.strip() for line in f.readlines() if line.strip() != ""][1:]
|
|
return md
|
|
|
|
|
|
def write_md(results):
|
|
"""
|
|
Write to README.md
|
|
"""
|
|
with open("../dummy.md", "w") as f:
|
|
f.write(results)
|
|
|
|
|
|
def _file_stem(f):
|
|
return f[-5:]
|
|
|
|
|
|
def get_problems(path=None, ext=None):
|
|
wd = os.getcwd()
|
|
os.chdir(Path(path))
|
|
scripts = sorted(
|
|
[script.as_posix() for script in Path(".").rglob(f"*[0-9].{ext}")],
|
|
key=_file_stem,
|
|
)
|
|
os.chdir(wd)
|
|
return scripts
|
|
|
|
|
|
def timing_Python(path=None, exec_only=1):
|
|
wd = os.getcwd()
|
|
scripts = get_problems(path=path, ext="py")
|
|
os.chdir(Path(path))
|
|
python_timings, python_results = [], []
|
|
for script in scripts[:exec_only]:
|
|
print(f"Executing {script}")
|
|
_res = run(
|
|
["python3", f"{script}"], capture_output=True, text=True
|
|
).stdout.split("\n")
|
|
python_timings.append(_res[0])
|
|
python_results.append(_res[2])
|
|
|
|
os.chdir(wd)
|
|
return python_timings, python_results
|
|
|
|
|
|
def timing_Julia(path=None, exec_only=1):
|
|
wd = os.getcwd()
|
|
scripts = get_problems(path="Julia", ext="jl")
|
|
os.chdir(Path(path))
|
|
julia_problem_numbers, julia_timings, julia_results = [], [], []
|
|
for script in scripts[:exec_only]:
|
|
print(f"Executing {script}")
|
|
_res = run(["julia", f"{script}"], capture_output=True, text=True).stdout.split(
|
|
"\n"
|
|
)
|
|
julia_problem_numbers.append(_res[0])
|
|
julia_timings.append(_res[1].split("(")[0])
|
|
julia_results.append(_res[3])
|
|
|
|
os.chdir(wd)
|
|
return julia_problem_numbers, julia_timings, julia_results
|
|
|
|
|
|
def execute(combine_columns=True, nproblems=-1):
|
|
"""
|
|
Execute scripts and return md formatted table
|
|
"""
|
|
python_timings, python_results = timing_Python(path="Python", exec_only=nproblems)
|
|
julia_n_problem, julia_timings, julia_results = timing_Julia(
|
|
path="Julia", exec_only=nproblems
|
|
)
|
|
|
|
assert len(python_timings) == len(julia_timings)
|
|
|
|
for i, (pr, jr) in enumerate(zip(python_results, julia_results), start=1):
|
|
try:
|
|
assert int(pr[22:]) == int(jr[23:])
|
|
except ValueError:
|
|
print(f"Results for problem {i} are not identical!\n")
|
|
continue
|
|
|
|
if combine_columns:
|
|
headers = [
|
|
"Problem #",
|
|
"Result",
|
|
"T_exec (Python)",
|
|
"T_exec (Julia)",
|
|
]
|
|
value_matrix = [
|
|
[f"{int(t_python[25:28]):03d}", int(r_python[22:]), t_python[29:], t_julia]
|
|
for t_python, r_python, t_julia in zip(
|
|
python_timings, python_results, julia_timings
|
|
)
|
|
]
|
|
writer_comb = MarkdownTableWriter(
|
|
table_name="Project Euler Solutions",
|
|
headers=headers,
|
|
value_matrix=value_matrix,
|
|
column_styles=[
|
|
Style(align="center"),
|
|
Style(align="right", thousand_separator="_"),
|
|
Style(align="right"),
|
|
Style(align="right"),
|
|
], # specify styles for each column
|
|
margin=1, # add a whitespace for both sides of each cell
|
|
)
|
|
writer_comb.write_table()
|
|
writer_comb.write_null_line()
|
|
|
|
else:
|
|
languages = ["Python", "Julia"]
|
|
|
|
for lang in languages:
|
|
if lang == "Python":
|
|
headers = ["Problem #", "Result", "T_exec (Python)"]
|
|
value_matrix = [
|
|
[f"{int(t_python[25:28]):03d}", int(r_python[22:]), t_python[29:]]
|
|
for t_python, r_python in zip(python_timings, python_results)
|
|
]
|
|
elif lang == "Julia":
|
|
headers = ["Problem #", "Result", "T_exec (Julia)"]
|
|
value_matrix = [
|
|
[f"{int(p_julia[-5:-1]):03d}", int(r_julia[23:]), t_julia]
|
|
for p_julia, r_julia, t_julia in zip(
|
|
julia_n_problem, julia_results, julia_timings
|
|
)
|
|
]
|
|
|
|
writer = MarkdownTableWriter(
|
|
table_name=f"Project Euler Solutions for {lang}",
|
|
headers=headers,
|
|
value_matrix=value_matrix,
|
|
column_styles=[
|
|
Style(align="center"),
|
|
Style(align="right", thousand_separator="_"),
|
|
Style(align="right"),
|
|
], # specify styles for each column
|
|
margin=1, # add a whitespace for both sides of each cell
|
|
)
|
|
writer.write_table()
|
|
writer.write_null_line()
|
|
# print(writer.dumps())
|
|
# return writer.dumps()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
md = read_md()
|
|
# print(md)
|
|
res = execute(combine_columns=True)
|
|
# print(res)
|
|
# write_md(res)
|