import unittest
import meshvis

class IndexedAttributes(unittest.TestCase):
    
    quad_verts = [[0, 0, 0], [1, 0, 0], [1, 1, 0], [0, 1, 0]]
    quad_faces_tri = [[0, 1, 2], [0, 2, 3]]

    def point_attribute_tri(self):
        attrib = meshvis.PointAttribute(['a', 'b', 'c', 'd'])
        resolved_attrib = meshvis.resolve_attributes(self.quad_faces_tri, [attrib])[0]
        self.assertEqual(len(resolved_attrib), 6)
        self.assertListEqual(resolved_attrib, ['a', 'b', 'c', 'a', 'c', 'd'])

    def face_attribute_tri(self):
        attrib = meshvis.FaceAttribute(['a', 'b'])
        resolved_attrib = meshvis.resolve_attributes(self.quad_faces_tri, [attrib])[0]
        self.assertEqual(len(resolved_attrib), 2)
        self.assertListEqual(resolved_attrib, ['a', 'b'])


if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(IndexedAttributes)
    unittest.TextTestRunner(verbosity=2).run(suite)