# Entries are (id, factory, expected_cy_dimension, expected_no_divs, expected_disc,
# expected_intersection_numbers_CY), where expected_disc is a list of
# (str(discriminant_factor), codimension) pairs as returned by ToricPolytope.disc().

PLAYGROUND_MODEL_FACTORIES = [
    (
        "K3_two_parameter_family",
        lambda: ToricPolytopeProjectiveSpace([1, 1, 2, 4], model_name="K3_two_parameter_family"),
        2,  # cy_dimension
        2,  # no_divs
        [("z2 - 1/4", 2), ("4096*z1^2*(4*z2 - 1) + 128*z1 - 1", 0)],
        {(0, 0): 4, (0, 1): 2, (1, 1): 0},
    ),
    (
        "CY3_two_parameter_family",
        lambda: ToricPolytopeProjectiveSpace([1, 1, 2, 2, 2], model_name="CY3_two_parameter_family"),
        3,
        2,
        [("z2 - 1/4", 3), ("65536*z1^2*(4*z2 - 1) + 512*z1 - 1", 0)],
        {(0, 0, 0): 8, (0, 0, 1): 4, (0, 1, 1): 0, (1, 1, 1): 0},
    ),
    (
        "CY4_two_parameter_family",
        lambda: ToricPolytopeProjectiveSpace([1, 1, 1, 1, 8, 12], model_name="CY4_two_parameter_family"),
        4,
        2,
        [
            ("z2 - 1/256", 2),
            ("34828517376*z1^4*(256*z2 - 1) + 322486272*z1^3 - 1119744*z1^2 + 1728*z1 - 1", 0),
        ],
        {(0, 0, 0, 0): 64, (0, 0, 0, 1): 16, (0, 0, 1, 1): 4, (0, 1, 1, 1): 1, (1, 1, 1, 1): 0},
    ),
    (
        "CICY3_one_parameter",
        lambda: ToricPolytopeCICY([[3, 3]], model_name="CICY3_one_parameter"),
        3,
        1,
        [("z1 - 1/46656", 0)],
        {(0, 0, 0): 9},
    ),
    (
        "CICY3_two_parameter",
        lambda: ToricPolytopeCICY([[3], [3]], model_name="CICY3_two_parameter"),
        3,
        2,
        [
            (
                "-19683*z1^3 - 2187*(27*z1 + 1)*z2^2 - 19683*z2^3 - 2187*z1^2 "
                "- 81*(729*z1^2 - 189*z1 + 1)*z2 - 81*z1 - 1",
                0,
            )
        ],
        {(0, 0, 0): 0, (0, 0, 1): 3, (0, 1, 1): 3, (1, 1, 1): 0},
    ),
    (
        "CICY5_two_parameter",
        lambda: ToricPolytopeCICY([[6, 1], [0, 2]], model_name="CICY5_two_parameter"),
        5,
        2,
        [
            (
                "16384*z2^7 - 28672*z2^6 + 21504*z2^5 - 448*(1647086*z1 - 5)*z2^3 - 8960*z2^4 "
                "- 112*(8235430*z1 + 3)*z2^2 - 678223072849*z1^2 - 28*(4941258*z1 - 1)*z2 - 1647086*z1 - 1",
                0,
            )
        ],
        {
            (0, 0, 0, 0, 0): 6,
            (0, 0, 0, 0, 1): 12,
            (0, 0, 0, 1, 1): 0,
            (0, 0, 1, 1, 1): 0,
            (0, 1, 1, 1, 1): 0,
            (1, 1, 1, 1, 1): 0,
        },
    ),
    (
        "CICY3_two_parameter_manual",
        lambda: ToricPolytope(
            [
                [1, 0, 0, 0, 0, 0],
                [0, 1, 0, 0, 0, 0],
                [0, 0, 1, 0, 0, 0],
                [0, 0, 0, 1, 0, 0],
                [0, 0, 0, 0, 1, 0],
                [0, 0, 0, 0, 0, 1],
                [-1, -1, 0, 0, 0, 0],
                [0, 0, -1, -1, -1, -1],
            ],
            nef_partition=[[0, 1, 2, 3], [4, 5, 6, 7]],
            model_name="CICY3_two_parameter_manual",
        ),
        4,
        2,
        [
            (
                "-14348907*z1^5 - 2657205*z1^4 - 196830*z1^3 - 29296875*(270*z1 + 1)*z2^2 "
                "- 30517578125*z2^3 - 7290*z1^2 + 9375*(98415*z1^3 - 32805*z1^2 + 810*z1 - 1)*z2 "
                "- 135*z1 - 1",
                0,
            )
        ],
        {(0, 0, 0, 0): 0, (0, 0, 0, 1): 0, (0, 0, 1, 1): 6, (0, 1, 1, 1): 8, (1, 1, 1, 1): 2},
    ),
]