review fix

Author: mgovindhasamy

hyperwallet/api.py

@@ -2809,7 +2809,7 @@ def createUserStatusTransition(self,
         :param data:
             A dictionary containing User Status Create Data information. **REQUIRED**
         :returns:
-            CreateUserStatusTransition.
+            StatusTransition.
         '''
         if not userToken:
             raise HyperwalletException('userToken is required')
@@ -2834,7 +2834,7 @@ def activateUser(self,
          :param userToken:
              A token identifying the User Create UserToken. **REQUIRED**
          :returns:
-             CreateUserStatusTransition.
+             StatusTransition.
          '''
         if not userToken:
             raise HyperwalletException('userToken is required')
@@ -2856,7 +2856,7 @@ def deactivateUser(self,
          :param userToken:
              A token identifying the User Create UserToken. **REQUIRED**
          :returns:
-             CreateUserStatusTransition.
+             StatusTransition.
          '''
         if not userToken:
             raise HyperwalletException('userToken is required')
@@ -2878,7 +2878,7 @@ def preactivateUser(self,
          :param userToken:
              A token identifying the User Create UserToken. **REQUIRED**
          :returns:
-             CreateUserStatusTransition.
+             StatusTransition.
          '''
         if not userToken:
             raise HyperwalletException('userToken is required')
@@ -2899,7 +2899,7 @@ def freezeUser(self,
          :param userToken:
              A token identifying the User Create UserToken. **REQUIRED**
          :returns:
-             CreateUserStatusTransition.
+             StatusTransition.
          '''
         if not userToken:
             raise HyperwalletException('userToken is required')
@@ -2921,7 +2921,7 @@ def lockUser(self,
          :param userToken:
              A token identifying the User Create UserToken. **REQUIRED**
          :returns:
-             CreateUserStatusTransition.
+             StatusTransition.
          '''
         if not userToken:
             raise HyperwalletException('userToken is required')

hyperwallet/tests/test_models.py

@@ -23,8 +23,7 @@
     StatusTransition,
     TransferMethodConfiguration,
     Webhook,
-    TransferRefunds,
-    CreateUserStatusTransition
+    TransferRefunds
 )
 
 
@@ -555,28 +554,5 @@ def test_transfer_refunds(self):
             )
         )
 
-
-    '''
-    
-    Create User Status Transition
-
-    '''
-
-    def test_create_user_status_transition(self):
-
-        status_transfer_refunds_data = {
-            "transition" : "LOCKED"
-        }
-
-        status_create_user_status_transition = CreateUserStatusTransition(status_transfer_refunds_data)
-
-        self.assertEqual(
-            status_create_user_status_transition.__repr__(),
-            'CreateUserStatusTransition({date}, {token})'.format(
-                date=status_transfer_refunds_data.get('createdOn'),
-                token=status_transfer_refunds_data.get('token')
-            )
-        )
-
 if __name__ == '__main__':
     unittest.main()