from midas import (
    MidasR,
    hash,
)
import csv

if __name__ == "__main__":
    midas = MidasR()
    with open("in.csv") as infile, open("test.out.csv", "w") as outfile:
        reader = csv.reader(infile)
        writer = csv.writer(outfile)
        for line in reader:
            source = hash(line[0])
            dest = hash(line[1])
            time = int(line[2])
            writer.writerow([
                line[0],
                line[1],
                time,
                "{:.6f}".format(midas.insert(source, dest, time)),
            ])