def test_search_single_item(self):
with open('tests/example_files/test_mocapbank.bvh') as f:
mocap = Bvh(f.read())
self.assertEqual([str(node) for node in mocap.search('ROOT')],
['ROOT Hips'])
def test_search_single_item_joints(self):
with open('tests/example_files/test_mocapbank.bvh') as f:
mocap = Bvh(f.read())
self.assertEqual(len(mocap.search('JOINT')), 18)
def test_search(self):
with open('tests/example_files/test_mocapbank.bvh') as f:
mocap = Bvh(f.read())
self.assertEqual(
[str(node) for node in mocap.search('JOINT', 'LeftShoulder')],
['JOINT LeftShoulder'])
