Philipp/aki_prj23_transparenzregister
1
0
Fork 0
mirror of https://github.com/fhswf/aki_prj23_transparenzregister.git synced 2025-06-22 08:43:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Moved AI files in the AI module. (#308)

Browse Source
This commit is contained in:
Philipp Horstenkamp
2023-11-02 20:30:04 +01:00
committed by GitHub
parent 3cc94cb652
commit 8d9981d967
12 changed files with 37 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/utils/mongo/ner_pipeline_test.py
View File

@ -4,8 +4,8 @@ from unittest.mock import Mock, patch
import pytest
from aki_prj23_transparenzregister.ai.ner_pipeline import EntityPipeline
from aki_prj23_transparenzregister.config.config_template import MongoConnection
from aki_prj23_transparenzregister.utils.mongo.ner_pipeline import EntityPipeline
@pytest.fixture()
@ -52,16 +52,14 @@ def mock_spacy(mocker: Mock) -> Mock:
"""
mock = Mock()
mocker.patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.init_spacy",
"aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.init_spacy",
return_value=mock,
)
return mock
# Mocking the NerAnalysisService methods
@patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.ner_spacy"
)
@patch("aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.ner_spacy")
def test_entity_pipeline_with_spacy(
mock_ner_spacy: Mock,
mock_mongo_connector: Mock,
@ -99,9 +97,7 @@ def test_entity_pipeline_with_spacy(
)
@patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.ner_spacy"
)
@patch("aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.ner_spacy")
def test_entity_pipeline_with_spacy_no_docs(
mock_ner_spacy: Mock,
mock_mongo_connector: Mock,
@ -137,7 +133,7 @@ def test_entity_pipeline_with_spacy_no_docs(
@patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.ner_company_list"
"aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.ner_company_list"
)
def test_entity_pipeline_with_companylist_ner(
mock_ner_companylist: Mock,
@ -178,7 +174,7 @@ def test_entity_pipeline_with_companylist_ner(
@patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.ner_company_list"
"aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.ner_company_list"
)
def test_entity_pipeline_with_companylist_ner_no_docs(
mock_ner_companylist: Mock,
@ -214,9 +210,7 @@ def test_entity_pipeline_with_companylist_ner_no_docs(
# Add more test cases for other NER methods (e.g., use_companylist_ner, use_transformer_ner) following a similar pattern.
@patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.ner_spacy"
)
@patch("aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.ner_spacy")
def test_entity_pipeline_with_transformer(
mock_ner_transformer: Mock,
mock_mongo_connector: Mock,
@ -254,9 +248,7 @@ def test_entity_pipeline_with_transformer(
)
@patch(
"aki_prj23_transparenzregister.utils.mongo.ner_service.NerAnalysisService.ner_spacy"
)
@patch("aki_prj23_transparenzregister.ai.ner_service.NerAnalysisService.ner_spacy")
def test_entity_pipeline_with_transformer_no_docs(
mock_ner_transformer: Mock,
mock_mongo_connector: Mock,

2
tests/utils/mongo/ner_service_test.py
View File

@ -1,6 +1,6 @@
"""Tests for checking NER Services."""
from aki_prj23_transparenzregister.utils.mongo.ner_service import NerAnalysisService
from aki_prj23_transparenzregister.ai.ner_service import NerAnalysisService
def test_ner_spacy() -> None:

14
tests/utils/mongo/sentiment_pipeline_test.py
View File

@ -4,10 +4,10 @@ from unittest.mock import Mock, patch
import pytest
from aki_prj23_transparenzregister.config.config_template import MongoConnection
from aki_prj23_transparenzregister.utils.mongo.sentiment_pipeline import (
from aki_prj23_transparenzregister.ai.sentiment_pipeline import (
SentimentPipeline,
)
from aki_prj23_transparenzregister.config.config_template import MongoConnection
@pytest.fixture()
@ -54,14 +54,14 @@ def mock_spacy(mocker: Mock) -> Mock:
"""
mock = Mock()
mocker.patch(
"aki_prj23_transparenzregister.utils.mongo.sentiment_service.SentimentAnalysisService.init_spacy",
"aki_prj23_transparenzregister.ai.sentiment_service.SentimentAnalysisService.init_spacy",
return_value=mock,
)
return mock
@patch(
"aki_prj23_transparenzregister.utils.mongo.sentiment_service.SentimentAnalysisService.sentiment_spacy"
"aki_prj23_transparenzregister.ai.sentiment_service.SentimentAnalysisService.sentiment_spacy"
)
def test_sentiment_pipeline_existing_sentiment(
mock_sentiment_spacy: Mock,
@ -102,7 +102,7 @@ def test_sentiment_pipeline_existing_sentiment(
@patch(
"aki_prj23_transparenzregister.utils.mongo.sentiment_service.SentimentAnalysisService.sentiment_spacy"
"aki_prj23_transparenzregister.ai.sentiment_service.SentimentAnalysisService.sentiment_spacy"
)
def test_sentiment_pipeline_no_documents(
mock_sentiment_spacy: Mock,
@ -134,7 +134,7 @@ def test_sentiment_pipeline_no_documents(
@patch(
"aki_prj23_transparenzregister.utils.mongo.sentiment_service.SentimentAnalysisService.sentiment_spacy"
"aki_prj23_transparenzregister.ai.sentiment_service.SentimentAnalysisService.sentiment_spacy"
)
def test_sentiment_pipeline_with_spacy(
mock_sentiment_spacy: Mock,
@ -173,7 +173,7 @@ def test_sentiment_pipeline_with_spacy(
# Mocking the SentimentAnalysisService methods
@patch(
"aki_prj23_transparenzregister.utils.mongo.sentiment_service.SentimentAnalysisService.sentiment_transformer"
"aki_prj23_transparenzregister.ai.sentiment_service.SentimentAnalysisService.sentiment_transformer"
)
def test_sentiment_pipeline_with_transformer(
mock_sentiment_transformer: Mock,

2
tests/utils/mongo/sentiment_service_test.py
View File

@ -1,7 +1,7 @@
"""Tests for checking Sentiment Services."""
from aki_prj23_transparenzregister.utils.mongo.sentiment_service import (
from aki_prj23_transparenzregister.ai.sentiment_service import (
SentimentAnalysisService,
)