45 lines
1.0 KiB
Python
45 lines
1.0 KiB
Python
|
import json
|
||
|
import os
|
||
|
import matplotlib.pyplot as plt
|
||
|
import subprocess
|
||
|
import csv
|
||
|
from tqdm import tqdm
|
||
|
|
||
|
|
||
|
# Run your code:
|
||
|
# Make sure that your code is using print_results_for_python
|
||
|
#executable_filename = ['make', 'run']
|
||
|
|
||
|
# Run reference code:
|
||
|
executable_filename = ['make', 'run-reference']
|
||
|
|
||
|
num_runs = 100
|
||
|
|
||
|
graph_repo = "data"
|
||
|
os.makedirs(graph_repo, exist_ok=True)
|
||
|
|
||
|
fancy_num_runs = range(0, num_runs, 1)
|
||
|
for run_id in tqdm(fancy_num_runs):
|
||
|
p = subprocess.run(executable_filename,
|
||
|
stdout=subprocess.PIPE, universal_newlines=True)
|
||
|
|
||
|
reader = csv.reader(p.stdout.splitlines())
|
||
|
|
||
|
dict_of_lists = dict()
|
||
|
|
||
|
for store_lev in range(1, 4+1, 1):
|
||
|
dict_of_lists[store_lev] = []
|
||
|
|
||
|
store_level = 1
|
||
|
for row in reader:
|
||
|
temp = row[0].split(' ')[:-1]
|
||
|
dict_of_lists[store_level] = list(map(int, temp))
|
||
|
store_level+=1
|
||
|
|
||
|
#print(dict_of_lists)
|
||
|
filename = graph_repo+"/run"+str(run_id)+".json"
|
||
|
jsonFile = open(filename, "w")
|
||
|
jsonFile.write(json.dumps(dict_of_lists))
|
||
|
jsonFile.close()
|
||
|
|