flake8 corrections

Author: chrsz

bvh.py

@@ -27,11 +27,11 @@ def __getitem__(self, key):
         for child in self.children:
             for index, item in enumerate(child.value):
                 if item == key:
-                    if index+1 >= len(child.value):
+                    if index + 1 >= len(child.value):
                         return None
                     else:
-                        return child.value[index+1:]
-        raise IndexError('key {0} not found'.format(key))
+                        return child.value[index + 1:]
+        raise IndexError('key {} not found'.format(key))
 
     def __repr__(self):
         return str(' '.join(self.value))

tests/test_bvh.py

@@ -32,6 +32,7 @@ def test_bones(self):
         bones = []
         with open('tests/test_freebvh.bvh') as f:
             mocap = Bvh(f.read())
+
         def iterate_joints(joint):
             bones.append(str(joint))
             for child in joint.filter('JOINT'):
@@ -118,7 +119,7 @@ def test_frame_channel(self):
         self.assertEqual(mocap.frame_joint_channel(22, 'Neck', 'Xrotation'), -6.77)
         self.assertEqual(mocap.frame_joint_channel(22, 'Head', 'Yrotation'), 8.47)
 
-    def test_frame_channel(self):
+    def test_frame_channel2(self):
         with open('tests/test_freebvh.bvh') as f:
             mocap = Bvh(f.read())
         self.assertEqual(mocap.frame_joint_channel(22, 'mixamorig:Hips', 'Xposition'), 4.3314)
@@ -128,7 +129,7 @@ def test_frame_iteration(self):
             mocap = Bvh(f.read())
         x_accumulator = 0.0
         for i in range(0, mocap.nframes):
-            x_accumulator += mocap.frame_joint_channel(i, 'Hips', 'Xposition') 
+            x_accumulator += mocap.frame_joint_channel(i, 'Hips', 'Xposition')
         self.assertTrue(abs(-19735.902699999995 - x_accumulator) < 0.0001)
 
     def test_joints_names(self):
@@ -151,16 +152,15 @@ def test_joint_parent(self):
     def test_frame_joint_multi_channels(self):
         with open('tests/test_mocapbank.bvh') as f:
             mocap = Bvh(f.read())
-        rotation = mocap.frame_joint_channels(30, 'Head', ['Xrotation', 'Yrotation', 'Zrotation']) 
+        rotation = mocap.frame_joint_channels(30, 'Head', ['Xrotation', 'Yrotation', 'Zrotation'])
         self.assertEqual(rotation, [1.77, 13.94, -7.42])
 
     def test_frames_multi_channels(self):
         with open('tests/test_mocapbank.bvh') as f:
             mocap = Bvh(f.read())
-        rotations = mocap.frames_joint_channels('Head', ['Xrotation', 'Yrotation', 'Zrotation']) 
+        rotations = mocap.frames_joint_channels('Head', ['Xrotation', 'Yrotation', 'Zrotation'])
         self.assertEqual(len(rotations), mocap.nframes)
 
 
 if __name__ == '__main__':
     unittest.main()
-