from .base import SymmetricSpectrum, GraphData

data = {
    3: GraphData(
        vertex_count=6,
        edge_count=6,
        distance_classes=[1, 2, 2, 1],
        spectrum=SymmetricSpectrum([(1, 2), (2, 1)]),
    ),
    4: GraphData(
        vertex_count=24,
        edge_count=36,
        distance_classes=[1, 3, 6, 9, 5],
        spectrum=SymmetricSpectrum(
            [
                (0, 4),
                (1, 3),
                (2, 6),
                (3, 1),
            ]
        ),
    ),
    5: GraphData(
        vertex_count=120,
        edge_count=240,
        distance_classes=[1, 4, 12, 30, 44, 26, 3],
        spectrum=SymmetricSpectrum(
            [
                (0, 30),
                (1, 4),
                (2, 28),
                (3, 12),
                (4, 1),
            ]
        ),
    ),
    6: GraphData(
        vertex_count=720,
        edge_count=1800,
        distance_classes=[1, 5, 20, 70, 170, 250, 169, 35],
        spectrum=SymmetricSpectrum(
            [
                (0, 168),
                (1, 30),
                (2, 120),
                (3, 105),
                (4, 20),
                (5, 1),
            ]
        ),
    ),
    7: GraphData(
        vertex_count=5040,
        edge_count=15120,
        distance_classes=[1, 6, 30, 135, 460, 1110, 1689, 1254, 340, 15],
        spectrum=SymmetricSpectrum(
            [
                (0, 840),
                (1, 468),
                (2, 495),
                (3, 830),
                (4, 276),
                (5, 30),
                (6, 1),
            ]
        ),
    ),
    8: GraphData(
        vertex_count=40320,
        spectrum=SymmetricSpectrum(
            [
                (0, 3960),
                (1, 5691),
                (2, 2198),
                (3, 6321),
                (4, 3332),
                (5, 595),
                (6, 42),
                (7, 1),
            ]
        ),
    ),
}

# TODO (maybe): verify these quantities with outputs from Haskell