Removed direct import and use package import. (#78)

Author: Kevin Hellemun

tests/bunq_test.py

@@ -4,7 +4,6 @@
 
 from bunq.sdk import context, util
 from bunq.sdk.client import ApiClient
-from bunq.sdk.context import BunqContext
 from bunq.sdk.exception import BunqException
 from bunq.sdk.model.generated import endpoint, object_
 from tests import config
@@ -52,13 +51,13 @@ class BunqSdkTestCase(unittest.TestCase):
 
     @classmethod
     def setUpClass(cls):
-        BunqContext.load_api_context(cls._get_api_context())
+        context.BunqContext.load_api_context(cls._get_api_context())
 
     def setUp(self):
         self.__set_second_monetary_account()
         self.__request_spending_money()
         time.sleep(0.5)
-        BunqContext.user_context().refresh_user_context()
+        context.BunqContext.user_context().refresh_user_context()
 
     def __set_second_monetary_account(self):
         response = endpoint.MonetaryAccountBank.create(
@@ -161,12 +160,12 @@ def alias_first(self):
         :rtype: Pointer
         """
 
-        if BunqContext.user_context().is_only_user_company_set():
-            return BunqContext.user_context().user_company.alias[
+        if context.BunqContext.user_context().is_only_user_company_set():
+            return context.BunqContext.user_context().user_company.alias[
                 self._FIRST_INDEX]
 
-        if BunqContext.user_context().is_only_user_person_set():
-            return BunqContext.user_context().user_person.alias[
+        if context.BunqContext.user_context().is_only_user_person_set():
+            return context.BunqContext.user_context().user_person.alias[
                 self._FIRST_INDEX]
 
         raise BunqException(self.__ERROR_COULD_NOT_DETERMINE_USER)

tests/model/generated/endpoint/test_attachment_public.py

@@ -1,6 +1,7 @@
+import os
+
 from bunq.sdk.client import ApiClient
-from bunq.sdk.model.generated.endpoint import AttachmentPublic
-from bunq.sdk.model.generated.endpoint import AttachmentPublicContent
+from bunq.sdk.model.generated import endpoint
 from tests.bunq_test import BunqSdkTestCase
 
 
@@ -24,10 +25,10 @@ def test_file_upload_and_retrieval(self):
                 self._ATTACHMENT_DESCRIPTION,
         }
 
-        attachment_uuid = AttachmentPublic.create(self.attachment_contents,
+        attachment_uuid = endpoint.AttachmentPublic.create(self.attachment_contents,
                                                   custom_headers).value
 
-        contents_from_response = AttachmentPublicContent.list(
+        contents_from_response = endpoint.AttachmentPublicContent.list(
             attachment_uuid).value
 
         self.assertEqual(self.attachment_contents, contents_from_response)