Philipp/aki_prj23_transparenzregister
1
0
Fork 0
mirror of https://github.com/fhswf/aki_prj23_transparenzregister.git synced 2025-12-17 20:10:45 +01: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

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,