import pytest
from mat3ra.made.material import Material
from mat3ra.made.tools.build.pristine_structures.two_dimensional.nanoribbon import create_nanoribbon
from mat3ra.made.tools.build_components.entities.reusable.one_dimensional.crystal_lattice_lines import (
    CrystalLatticeLinesBuilder,
    CrystalLatticeLinesConfiguration,
)

from .fixtures.monolayer import GRAPHENE
from .fixtures.nanoribbon.armchair import GRAPHENE_NANORIBBON_ARMCHAIR
from .fixtures.nanoribbon.zigzag import GRAPHENE_NANORIBBON_ZIGZAG
from .utils import assert_two_entities_deep_almost_equal


@pytest.mark.parametrize(
    "material_config, width, length, vacuum_width, vacuum_length, miller_indices_2d, edge_type, expected_nanoribbon",
    [
        (GRAPHENE, 5, 4, 5.0, 5.0, (1, 1), "armchair", GRAPHENE_NANORIBBON_ARMCHAIR),
        (GRAPHENE, 2, 4, 5.0, 5.0, (0, 1), "zigzag", GRAPHENE_NANORIBBON_ZIGZAG),
    ],
)
def test_build_nanoribbon(
    material_config, width, length, vacuum_width, vacuum_length, miller_indices_2d, edge_type, expected_nanoribbon
):
    material = Material.create(material_config)

    nanoribbon = create_nanoribbon(
        material=material,
        miller_indices_2d=miller_indices_2d,
        width=width,
        length=length,
        vacuum_width=vacuum_width,
        vacuum_length=vacuum_length,
    )

    assert_two_entities_deep_almost_equal(nanoribbon, expected_nanoribbon)


def test_crystal_lattice_lines():
    builder = CrystalLatticeLinesBuilder()
    config = CrystalLatticeLinesConfiguration(crystal=Material.create(GRAPHENE), miller_indices_2d=(1, 0))

    material = builder.get_material(config)
    material.name = f"{material.name} {config.miller_indices_2d}"
    assert isinstance(material, Material)