from sage.all import *
import cProfile, pstats
import itertools
load('aplK.py')
load('minimalModel_parallel.py')
#######################################################
# TESTS

#S = simplicial_complexes.Sphere(2)
S = simplicial_complexes.K3Surface()
S.set_immutable()

def test():
    [sullivan_model, model_morphism, cohomology_correspondence] = minimal_model(S, max_depth=12000, max_dim=3)
    print(sullivan_model, flush=True)
    save(sullivan_model, "sullivan_model_K3_deg3")
    save(model_morphism,"model_morphism_K3_deg3")
    save(cohomology_correspondence,"cohomology_correspondence_K3_deg3")
#############################################
cProfile.runctx("test()", globals(), locals(), "Profile_K3_deg3.prof")
s = pstats.Stats("Profile_K3_deg3.prof")
s.strip_dirs().sort_stats("cumtime").print_stats()