andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

More experimentation

Browse source 
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-04-03 21:46:35 +02:00
parent 1c501db81a
commit 83d870d2ea
43 changed files with 749 additions and 496 deletions
Download patch file Download diff file Expand all files Collapse all files

2
good_ai/tests/open_s3/test_human_readable_to_byte.py
View file

@ -1,6 +1,6 @@
import unittest
from src.open_s3.helper import human_readable_to_byte
from src.good_ai.open_s3.helper import human_readable_to_byte
class TestHumanReadableToByte(unittest.TestCase):

12
good_ai/tests/open_s3/test_large_file.py
View file

@ -8,7 +8,7 @@ import botocore.session
PATH = Path(__file__).parent.resolve()
from src.open_s3 import LargeFile
from src.good_ai import LargeFile
credentials = {
"aws_region_name": "your_region_like_eu-west-2",
@ -30,7 +30,7 @@ class TestLargeFile(unittest.TestCase):
self.assertRaises(ValueError, LargeFile, "test-file", "test")
@patch("botocore.session")
def test_initialized_with_dict(self, session) -> None:
def test_initialized_with_dict(self, session: Any) -> None:
session_mock = Mock()
session.get_session = create_autospec(
botocore.session.get_session, return_value=session_mock
@ -58,7 +58,13 @@ class TestLargeFile(unittest.TestCase):
session_mock.create_client = Mock(return_value=s3)
LargeFile.configure_credentials(**credentials)
LargeFile.configure_credentials(
aws_region_name=credentials["aws_region_name"],
aws_access_key_id=credentials["aws_access_key_id"],
aws_secret_access_key=credentials["aws_secret_access_key"],
large_files_bucket_name=credentials["large_files_bucket_name"],
endpoint_url=credentials["endpoint_url"],
)
lf = LargeFile("test-file")
session_mock.set_credentials.assert_called_once_with(
access_key=credentials["aws_access_key_id"],