import pytest
from mat3ra.made.material import Material
from mat3ra.made.tools.build.defective_structures.two_dimensional.terrace.helpers import create_defect_terrace
from mat3ra.made.tools.build.pristine_structures.two_dimensional.slab.helpers import create_slab
from unit.fixtures.bulk import BULK_Si_CONVENTIONAL
from unit.fixtures.terrace import TERRACE_SLAB_Si_001_3x3
from unit.utils import assert_two_entities_deep_almost_equal


@pytest.mark.parametrize(
    "slab_parameters, cut_direction, pivot_coordinate, num_added_layers, expected_config",
    [
        (
            {"crystal": BULK_Si_CONVENTIONAL, "number_of_layers": 1, "xy_supercell_matrix": [[1, 0], [0, 2]]},
            [0, 1, 0],
            [0.5, 0.5, 0.5],
            1,
            TERRACE_SLAB_Si_001_3x3,
        )
    ],
)
def test_create_terrace(
    slab_parameters,
    cut_direction,
    pivot_coordinate,
    num_added_layers,
    expected_config,
):
    slab = create_slab(
        Material.create(slab_parameters["crystal"]),
        number_of_layers=slab_parameters["number_of_layers"],
        xy_supercell_matrix=slab_parameters["xy_supercell_matrix"],
        termination_top_formula=None,
        termination_bottom_formula=None,
    )
    terrace = create_defect_terrace(
        slab=slab,
        cut_direction=cut_direction,
        pivot_coordinate=pivot_coordinate,
        number_of_added_layers=num_added_layers,
        rotate_to_match_pbc=True,
    )

    assert_two_entities_deep_almost_equal(terrace, expected_config)