added actor component tests

Roberto De Ioris · Roberto De Ioris · commit 3ac5dda47bf2 · 2017-02-12T13:49:22.000+01:00
diff --git a/tests/test_actor.py b/tests/test_actor.py
@@ -1,6 +1,6 @@
 import unittest
 import unreal_engine as ue
-from unreal_engine.classes import Actor, Character
+from unreal_engine.classes import Actor, Character, CharacterMovementComponent, CapsuleComponent
 from unreal_engine import FVector, FRotator
 import time
 import math
@@ -44,5 +44,17 @@ def test_destroy(self):
     	found_actor = self.world.find_actor_by_label(actor_label)
     	self.assertIsNone(found_actor)
 
+    def test_has_component(self):
+    	new_actor = self.world.actor_spawn(Character, FVector(100, 200, 300))
+    	self.assertTrue(new_actor.actor_has_component_of_type(CapsuleComponent))
+
+    def test_get_component(self):
+    	new_actor = self.world.actor_spawn(Character, FVector(100, 200, 300))
+    	self.assertTrue(new_actor.get_actor_component_by_type(CharacterMovementComponent).bIsActive)
+
+    def test_get_components(self):
+    	new_actor = self.world.actor_spawn(Character, FVector(100, 200, 300))
+    	self.assertTrue(len(new_actor.get_actor_components()), 4)
+
 
 
diff --git a/tests/test_uobject.py b/tests/test_uobject.py
@@ -1,7 +1,6 @@
 import unittest
 import unreal_engine as ue
-from unreal_engine.classes import Material, Character
-from unreal_engine import FVector, FRotator
+from unreal_engine.classes import Material
 import time
 import math
 
