File size: 1,171 Bytes
cf8614b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | 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()
|