add AMPS (#585)

* add AMPS * fix issues * re-formatting
ElectronicBabylonianLiterature · Nov 29, 2024 · 86d160a · 86d160a
1 parent f0339d5
commit 86d160a
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 5 deletions.
diff --git a/ebl/common/domain/project.py b/ebl/common/domain/project.py
@@ -4,3 +4,4 @@
 class ResearchProject(NamedEnumWithParent):
     CAIC = ("Cuneiform Artefacts of Iraq in Context", "CAIC", None)
     ALU_GENEVA = ("Edition of the Omen Series Summa Alu", "aluGeneva", None)
+    AMPS = ("Ancient Mesopotamian Priestly Scholasticism", "AMPS", None)
diff --git a/ebl/tests/factories/fragment.py b/ebl/tests/factories/fragment.py
@@ -234,7 +234,7 @@ class Meta:
     introduction = Introduction("text", (StringPart("text"),))
     notes = Notes("notes", (StringPart("notes"),))
     external_numbers = factory.SubFactory(ExternalNumbersFactory)
-    projects = (ResearchProject.CAIC,)
+    projects = (ResearchProject.CAIC, ResearchProject.ALU_GENEVA, ResearchProject.AMPS)
     archaeology = factory.SubFactory(ArchaeologyFactory)
     colophon = factory.SubFactory(ColophonFactory)
 

diff --git a/ebl/tests/fragmentarium/test_dtos.py b/ebl/tests/fragmentarium/test_dtos.py
@@ -113,7 +113,11 @@ def expected_dto(lemmatized_fragment, has_photo):
             "externalNumbers": ExternalNumbersSchema().dump(
                 lemmatized_fragment.external_numbers
             ),
-            "projects": [ResearchProject["CAIC"].abbreviation],
+            "projects": [
+                ResearchProject["CAIC"].abbreviation,
+                ResearchProject["ALU_GENEVA"].abbreviation,
+                ResearchProject["AMPS"].abbreviation,
+            ],
             "archaeology": ArchaeologySchema().dump(lemmatized_fragment.archaeology),
             "colophon": ColophonSchema().dump(lemmatized_fragment.colophon),
             "authorizedScopes": [],

diff --git a/ebl/tests/fragmentarium/test_fragment_repository.py b/ebl/tests/fragmentarium/test_fragment_repository.py
@@ -1084,11 +1084,12 @@ def test_query_genres(fragment_repository, query, expected):
     [
         ("CAIC", [0]),
         ("aluGeneva", [1]),
-        (None, [0, 1]),
+        ("AMPS", [2]),
+        (None, [0, 1, 2]),
     ],
 )
 def test_query_project(fragment_repository, query, expected):
-    projects = [ResearchProject.CAIC, ResearchProject.ALU_GENEVA]
+    projects = [ResearchProject.CAIC, ResearchProject.ALU_GENEVA, ResearchProject.AMPS]
 
     fragments = [
         FragmentFactory.build(

diff --git a/ebl/tests/fragmentarium/test_fragments_search_route.py b/ebl/tests/fragmentarium/test_fragments_search_route.py
@@ -387,7 +387,7 @@ def test_search_script_period(client, fragmentarium, params, expected):
 
 @pytest.mark.parametrize(
     "project",
-    [ResearchProject.CAIC, ResearchProject.ALU_GENEVA],
+    [ResearchProject.CAIC, ResearchProject.ALU_GENEVA, ResearchProject.AMPS],
 )
 def test_search_project(client, fragmentarium, project):
     fragments = [