#
# Copyright 2022 Adobe. All rights reserved.
# This file is licensed to you under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License. You may obtain a copy
# of the License at http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed under
# the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
# OF ANY KIND, either express or implied. See the License for the specific language
# governing permissions and limitations under the License.
#
import lagrange

import numpy as np

from .assets import single_triangle, cube


class TestComputeVertexValence:
    def test_simple(self, single_triangle):
        mesh = single_triangle
        id = lagrange.compute_vertex_valence(mesh)
        attr = mesh.attribute(id)
        assert np.all(attr.data == 2)

    def test_cube(self, cube):
        mesh = cube
        id = lagrange.compute_vertex_valence(mesh)
        attr = mesh.attribute(id)
        assert np.all(attr.data == 3)

    def test_square(self):
        mesh = lagrange.SurfaceMesh()
        mesh.vertices = np.array([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]])
        mesh.facets = np.array([[0, 1, 2], [2, 1, 3]])

        id = lagrange.compute_vertex_valence(mesh)
        attr = mesh.attribute(id)
        assert np.all(attr.data == [2, 3, 3, 2])
