import unittest
from tests.core import TestCore
from pyrep.const import ObjectType
from pyrep.objects.camera import Camera
from pyrep.objects.dummy import Dummy
from pyrep.objects.object import Object
import numpy as np


class TestCameras(TestCore):

    def setUp(self):
        super().setUp()
        self.camera = Camera('DefaultCamera')
        self.dummy = Dummy('dummy')

    def test_create(self):
        with self.assertRaises(NotImplementedError):
            _ = Camera.create()

    def test_get_set_position(self):
        position = self.camera.get_position()
        self.assertIsInstance(position, np.ndarray)
        self.assertEqual(position.shape, (3,))

        self.camera.set_position([0.1, 0.1, 0.1], self.dummy)
        self.assertTrue(np.allclose(
            self.camera.get_position(self.dummy), [0.1, 0.1, 0.1]))
        self.camera.set_position([0.2, 0.2, 0.2])
        self.assertTrue(np.allclose(self.camera.get_position(), [0.2, 0.2, 0.2]))

    def test_get_object_type(self):
        self.assertEqual(Object.get_object_type('DefaultCamera'),
                         ObjectType.CAMERA)


if __name__ == '__main__':
    unittest.main()