More experimentation

Signed-off-by: András Schmelczer <andras@schmelczer.dev>

good_ai/tests/open_s3/test_human_readable_to_byte.py

@@ -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):

good_ai/tests/open_s3/test_large_file.py

@@ -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"],

good_ai/tests/sus/data/parsed.py

@@ -1,4 +1,4 @@
-from sus.publication_tei.models import (
+from good_ai.utilities.publication_tei.models import (
     Affiliation,
     Author,
     Meta,

good_ai/tests/sus/test_clean.py

@@ -1,6 +1,6 @@
 import unittest
 
-from src.sus.clean import clean
+from src.good_ai.utilities.clean import clean
 
 
 class TestClean(unittest.TestCase):

good_ai/tests/sus/test_evaluate_ranking.py

@@ -1,7 +1,7 @@
 import unittest
 from pathlib import Path
 
-from src.sus.evaluate_ranking import evaluate_ranking
+from src.good_ai.utilities.evaluate_ranking import evaluate_ranking
 
 
 class TestEvaluateRanking(unittest.TestCase):

good_ai/tests/sus/test_get_sentences.py

@@ -1,6 +1,6 @@
 import unittest
 
-from src.sus.get_sentences import get_sentences
+from src.good_ai.utilities.get_sentences import get_sentences
 
 
 class TestGetSentences(unittest.TestCase):

good_ai/tests/sus/test_language.py

@@ -1,6 +1,10 @@
 import unittest
 
-from src.sus.language import english_name_of_language, is_english, predict_language
+from src.good_ai.utilities.language import (
+    english_name_of_language,
+    is_english,
+    predict_language,
+)
 
 
 class TestLanguage(unittest.TestCase):

good_ai/tests/sus/test_lemmatize_text.py

@@ -1,6 +1,6 @@
 import unittest
 
-from src.sus.lemmatize_text import lemmatize_text
+from src.good_ai.utilities.lemmatize_text import lemmatize_text
 
 
 class TestLemmatizeText(unittest.TestCase):

good_ai/tests/sus/test_lemmatize_token.py

@@ -1,7 +1,7 @@
 import unittest
 
-from src.sus.lemmatize_text import lemmatize_token
-from src.sus.nlp import nlp
+from src.good_ai.utilities.lemmatize_text import lemmatize_token
+from src.good_ai.utilities.nlp import nlp
 
 
 class TestLemmatizeToken(unittest.TestCase):

good_ai/tests/sus/test_match_names.py

@@ -1,6 +1,6 @@
 import unittest
 
-from src.sus.match_names.match_names import match_names
+from src.good_ai.utilities.match_names.match_names import match_names
 
 
 class TestMatchNames(unittest.TestCase):

good_ai/tests/sus/test_parallel_map.py

@@ -1,6 +1,6 @@
 import unittest
 
-from src.sus.parallel_map import parallel_map
+from src.good_ai.utilities.parallel_map import parallel_map
 
 COUNT = int(1e5) + 3
 

good_ai/tests/sus/test_publication_tei.py

@@ -1,7 +1,7 @@
 import unittest
 from pathlib import Path
 
-from src.sus.publication_tei import PublicationTEI
+from src.good_ai.utilities.publication_tei import PublicationTEI
 
 from .data.parsed import authors, content, metadata, sentences
 

good_ai/tests/sus/test_unique.py

@@ -1,6 +1,6 @@
 import unittest
 
-from src.sus.unique import unique
+from src.good_ai.utilities.unique import unique
 
 original = [
     ("a", 1),