#!/usr/bin/env python

import unittest

from argaze.ArUcoMarkers import ArUcoMarkersDictionary, ArUcoMarker

class TestArUcoMarkerClass(unittest.TestCase):
    """Test ArUcoMarker class."""

    def test_new(self):
        """Test ArUcoMarker creation."""

        # Check DICT_ARUCO_ORIGINAL ArUcoMarker creation
        aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('DICT_ARUCO_ORIGINAL')

        aruco_marker = aruco_dictionary.create_marker(0, 3)

        self.assertEqual(aruco_marker.dictionary, aruco_dictionary)
        self.assertEqual(aruco_marker.identifier, 0)
        self.assertEqual(aruco_marker.size, 3)

if __name__ == '__main__':

    unittest.main()