diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_transform.py | 14 | ||||
| -rw-r--r-- | tests/test_work_pipeline.py | 4 |
2 files changed, 14 insertions, 4 deletions
diff --git a/tests/test_transform.py b/tests/test_transform.py index 4409bf0..fd14533 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -1,8 +1,16 @@ +import json + from fatcat_openapi_client import ReleaseEntity -from fatcat_scholar.api_entities import * -from fatcat_scholar.schema import * -from fatcat_scholar.transform import * +from fatcat_scholar.api_entities import entity_from_json +from fatcat_scholar.schema import ScholarBiblio +from fatcat_scholar.transform import ( + biblio_metadata_hacks, + es_biblio_from_release, + es_release_from_release, + run_refs, + run_transform, +) def test_es_release_from_release() -> None: diff --git a/tests/test_work_pipeline.py b/tests/test_work_pipeline.py index 554cc07..bc8a79a 100644 --- a/tests/test_work_pipeline.py +++ b/tests/test_work_pipeline.py @@ -1,3 +1,5 @@ +from typing import Any + import responses from fatcat_scholar.config import settings @@ -6,7 +8,7 @@ from fatcat_scholar.sandcrawler import ( SandcrawlerMinioClient, SandcrawlerPostgrestClient, ) -from fatcat_scholar.work_pipeline import * +from fatcat_scholar.work_pipeline import WorkPipeline @responses.activate |