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
|
import platform
import pytest
import numpy as np
import dartpy as dart
def test_basic():
urdfParser = dart.utils.DartLoader()
kr5 = urdfParser.parseSkeleton("dart://sample/urdf/KR5/KR5 sixx R650.urdf")
assert kr5 is not None
for i in range(kr5.getNumBodyNodes()):
body = kr5.getBodyNode(i)
assert np.array_equal(np.array(body.getSpatialVelocity()), np.zeros(6)) is True
shape_nodes = body.getShapeNodes()
for shape_node in shape_nodes:
print(shape_node)
if shape_node.hasVisualAspect():
visual = shape_node.getVisualAspect()
visual.getRGBA()
if shape_node.hasCollisionAspect():
collision = shape_node.getCollisionAspect()
if shape_node.hasDynamicsAspect():
dynamics = shape_node.getDynamicsAspect()
if __name__ == "__main__":
pytest.main()
|
