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'])