Feature update function to Deposit resource

starkbank/deposit/__deposit.py

@@ -125,9 +125,9 @@ def update(id, amount=None, user=None):
     """# Update Deposit entity
     Update the Deposit by passing its id to be partially or fully reversed.
     ## Parameters (required):
-    - id [string]: Deposit id. ex: '5656565656565656'
+    - id [string]: Deposit id. ex: "5656565656565656"
     ## Parameters (optional):
-    - amount [string]: The new amount of the Deposit. If the amount = 0 the Deposit will be fully reversed
+    - amount [string, default None]: The new amount of the Deposit. If the amount = 0 the Deposit will be fully reversed
     - user [Organization/Project object, default None]: Organization or Project object. Not necessary if starkbank.user was set before function call
     ## Return:
     - target Deposit with updated attributes

tests/sdk/test_deposit.py

@@ -47,21 +47,22 @@ def test_success(self):
         print(deposit)
 
 
-if __name__ == '__main__':
-    main()
-
-
 class TestDepositInfoPatch(TestCase):
 
     def test_success_amount(self):
         deposits = starkbank.deposit.query(status="created", limit=1)
         deposit_amount = 0
         for deposit in deposits:
             self.assertIsNotNone(deposit.id)
-            print(deposit)
             updated_deposit = starkbank.deposit.update(
                 deposit.id,
                 amount=deposit_amount,
             )
             print(updated_deposit)
             self.assertEqual(updated_deposit.amount, deposit_amount)
+
+
+if __name__ == '__main__':
+    main()
+
+