Adding uni test for cylinder_actor

Author: ManishReddyR

fury/actor.py

@@ -196,15 +196,15 @@ def cylinder(
     centers,
     colors,
     *,
-    height= 1,
-    sectors= 36,
-    radii= 0.5,
+    height=1,
+    sectors=36,
+    radii=0.5,
     scales=(1, 1, 1),
     directions=(0, 1, 0),
-    capped= True,
-    opacity= None,
-    material= "phong",
-    enable_picking= True,
+    capped=True,
+    opacity=None,
+    material="phong",
+    enable_picking=True,
 ):
     """Visualize one or many cylinders with different features.
 
@@ -259,7 +259,8 @@ def cylinder(
     >>> show_manager.start()
     """
 
-    vertices, faces = fp.prim_cylinder(radius= radii, height= height, sectors= sectors, capped= capped)
+    vertices, faces = fp.prim_cylinder(
+        radius=radii, height=height, sectors=sectors, capped=capped)
     res = fp.repeat_primitive(
         vertices,
         faces,
@@ -287,7 +288,8 @@ def cylinder(
         texcoords=big_vertices.astype("float32"),
         colors=big_colors.astype("float32"),
     )
-    mat = _create_mesh_material(material=material, enable_picking=enable_picking)
+    mat = _create_mesh_material(material=material,
+                                enable_picking=enable_picking)
     obj = create_mesh(geometry=geo, material=mat)
     obj.local.position = centers[0]
 

fury/tests/test_actor.py

@@ -1,6 +1,8 @@
 import numpy as np
 import numpy.testing as npt
 
+from PIL import Image
+
 from fury import actor, window
 
 
@@ -91,3 +93,66 @@ def test_box():
     assert box_actor.prim_count == 1
 
     scene.remove(box_actor)
+
+
+def test_cylinder():
+    scene = window.Scene()
+    centers= np.array([[0,0,0]])
+    colors= np.array([[1,0,0]])
+    sectors= 36
+    capped= True
+
+    cylinder_actor= actor.cylinder(
+        centers=centers, colors=colors, sectors=sectors, capped=capped)
+    scene.add(cylinder_actor)
+
+    npt.assert_array_equal(cylinder_actor.local.position, centers[0])
+    mean_vertex= np.mean(cylinder_actor.geometry.positions.view, axis=0)
+
+    npt.assert_array_almost_equal(mean_vertex, centers[0], decimal=2)
+    assert cylinder_actor.prim_count == 1
+
+    window.snapshot(scene=scene, fname="cylinder_test_1.png")
+
+    img = Image.open("cylinder_test_1.png")
+    img_array = np.array(img)
+
+    mean_r, mean_g, mean_b, mean_a = np.mean(
+        img_array.reshape(-1, img_array.shape[2]), axis=0
+    )
+
+    assert mean_r > mean_b and mean_r > mean_g
+    assert 0 < mean_r < 255 and 0 < mean_g < 255 and 0 <= mean_b < 255
+
+    middle_pixel = img_array[img_array.shape[0]//2, img_array.shape[1]//2]
+    r, g, b, a = middle_pixel
+    assert r > g and r > b
+    assert g == b
+    assert r > 0 and g > 0 and b > 0
+
+    scene.remove(cylinder_actor)
+
+    cylinder_actor_2 = actor.cylinder(
+        centers=centers, colors=colors, sectors=sectors, capped=capped, material='basic')
+    scene.add(cylinder_actor_2)
+    window.snapshot(scene=scene, fname="cylinder_test_2.png")
+
+    img = Image.open("cylinder_test_2.png")
+    img_array = np.array(img)
+
+    mean_r, mean_g, mean_b, mean_a = np.mean(
+        img_array.reshape(-1, img_array.shape[2]), axis=0
+    )
+
+    assert mean_r > mean_b and mean_r > mean_g
+    assert 0 < mean_r < 255
+    assert mean_g == 0 and mean_b == 0
+
+    middle_pixel = img_array[img_array.shape[0]//2,
+                              img_array.shape[1]//2]
+    r, g, b, a = middle_pixel
+    assert r > g and r > b
+    assert g == 0 and b == 0
+    assert r == 255
+
+    scene.remove(cylinder_actor_2)