def delete_all_directories(step):
x = rglob.rglob(world.root, "*")
x.sort(reverse=True)
for d in x:
if os.path.isdir(d):
os.rmdir(d)
else:
os.remove(d)
del world.dirs
world.dirs = []
del world.known_sizes
world.known_sizes = {}
def parseFiles():
# (bufsz, protocol, src, dest, flow_id, time, throughput)
res = []
# (bufsz, time, fairness)
res2 = []
for file_name in rglob.rglob(".", "*.txt"):
with open(file_name) as fp:
lines = fp.readlines()
bufsz = int(lines[0].split()[-2])
lines = lines[3:]
while len(lines) > 0:
while(len(lines[0].split()) == 0):
lines = lines[1:]
time = int(lines[0].split()[-2])
lines = lines[3:]
if time >= 35:
limit = 10
else:
limit = 8
for i in range(limit):
vals = lines[i].split()
flowid = int(vals[0])
protocol = vals[1]
src = vals[2]
dest = vals[3]
throughput = float(vals[4])
res.append((bufsz, protocol, src, dest, flowid, time, throughput))
lines = lines[limit + 1:]
fairness = float(lines[0].split()[-1])
res2.append((bufsz, time, fairness))
lines = lines[3:]
return res, res2
def find_files(step, expected_num_of_files, file_type):
x = rglob.rglob(world.root, "*{}".format(file_type))
assert len(x) == int(expected_num_of_files), \
"Found {} files".format(len(x))
def find_directories(step, expected_num_of_dirs):
x = rglob.rglob(world.root, "*_rglob")
assert len(x) == int(expected_num_of_dirs), \
"Found {} directories".format(len(x))
