import time
import sys
import jac_2d
import pickle

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print "supply dimension"
        sys.exit()
    dim = int(sys.argv[1])
    start = time.time()
    output = jac_2d.solver(dim)
    finish = time.time()
    print finish - start
    fd = open("data.txt", "w")
    pickle.dump(output, fd)
    fd.close()
def setup(dim):
    u = np.zeros((dim, dim), dtype=np.float32)
    initialize(u)
    u_new = np.copy(u)
    u_new = jac_2d.solver(dim, u, u_new)
    return u_new