예제 #1
0
 def test_check_standard_no_hip_exceed(self):
     l_hip = Point(60, 120, 1, 0, 0)
     l_knee = Point(62, 125, 1, 0, 0)
     r_hip = Point(40, 10, 1, 0, 0)
     r_knee = Point(42, 125, 1, 0, 0)
     result = self.rule.check_standard(l_hip, r_hip, l_knee, r_knee)
     self.assertFalse(result)
예제 #2
0
 def test_check_standard_both_hips_exceed(self):
     l_hip = Point(60, 126, 1, 0, 0)
     l_knee = Point(62, 125, 1, 0, 0)
     r_hip = Point(40, 126, 1, 0, 0)
     r_knee = Point(42, 125, 1, 0, 0)
     result = self.rule.check_standard(l_hip, r_hip, l_knee, r_knee)
     self.assertTrue(result)
예제 #3
0
 def setUp(self):
     self.points = [
         # x, y, conf, idx, z
         Point(50, 50, 1, 0, 0),     #     0: 'Nose',
         Point(50, 55, 1, 0, 0),     #     1: 'Neck',
         Point(40, 55, 1, 0, 0),     #     2: 'RShoulder',
         Point(35, 75, 1, 0, 0),     #     3: 'RElbow',
         Point(30, 95, 1, 0, 0),     #     4: 'RWrist',
         Point(60, 55, 1, 0, 0),     #     5: 'LShoulder',
         Point(65, 75, 1, 0, 0),     #     6: 'LElbow',
         Point(70, 95, 1, 0, 0),     #     7: 'LWrist',
         Point(43, 95, 1, 0, 0),     #     8: 'RHip',
         Point(42, 125, 1, 0, 0),    #     9: 'RKnee',
         Point(40, 155, 1, 0, 0),    #     10: 'RAnkle',
         Point(57, 95, 1, 0, 0),     #     11: 'LHip',
         Point(58, 125, 1, 0, 0),    #     12: 'LKnee',
         Point(60, 155, 1, 0, 0),    #     13: 'LAnkle',
         Point(48, 47, 1, 0, 0),     #     14: 'REye',
         Point(52, 47, 1, 0, 0),     #     15: 'LEye',
         Point(45, 48, 1, 0, 0),     #     16: 'REar',
         Point(55, 48, 1, 0, 0)      #     17: 'LEar'
     ]
예제 #4
0
 def test_get_part(self):
     result = get_part(self.points, 'LAnkle')
     l_ankle = Point(60, 155, 1, 0, 0)
     self.assertEqual(l_ankle, result)
예제 #5
0
 def test_right_knee_buckle(self):
     hip = Point(60, 95, 1, 0, 0)
     knee = Point(70, 125, 1, 0, 0)
     ankle = Point(60, 155, 1, 0, 0)
     result = self.rule.knee_buckle(hip, knee, ankle, 'right')
     self.assertTrue(result)
예제 #6
0
 def test_left_knee_not_buckle(self):
     hip = Point(60, 95, 1, 0, 0)
     knee = Point(58, 125, 1, 0, 0)
     ankle = Point(60, 155, 1, 0, 0)
     result = self.rule.knee_buckle(hip, knee, ankle, 'left')
     self.assertFalse(result)
예제 #7
0
 def test_knee_not_exceed_toe(self):
     # foot len is 17.12
     knee = Point(58, 125, 1, 0, 17)
     ankle = Point(60, 155, 1, 0, 0)
     result = self.rule.knee_exceed_toe(knee, ankle)
     self.assertFalse(result)
예제 #8
0
 def test_hip_same_height(self):
     l_hip = Point(57, 120, 1, 0, 0)
     r_hip = Point(47, 119.5, 1, 0, 0)
     result = self.rule.hip_different_height(l_hip, r_hip)
     self.assertFalse(result)
예제 #9
0
 def test_hip_higher_than_knee(self):
     hip = Point(57, 120, 1, 0, 0)
     knee = Point(58, 125, 1, 0, 0)
     result = self.rule.hip_lower_than_knee(hip, knee)
     self.assertFalse(result)
예제 #10
0
 def test_shoulder_same_height(self):
     l_shoulder = Point(60, 55, 1, 0, 0)
     r_shoulder = Point(40, 55, 1, 0, 0)
     result = self.rule.shoulder_different_height(l_shoulder, r_shoulder)
     self.assertFalse(result)
예제 #11
0
 def test_ankle_same_width(self):
     neck = Point(50, 55, 1, 0, 0)
     l_ankle = Point(60, 155, 1, 0, 0)
     r_ankle = Point(40.5, 155, 1, 0, 0)
     result = self.rule.ankle_different_width(neck, l_ankle, r_ankle)
     self.assertFalse(result)