Alex Galarce
55f96e77fc
Feature/contract to aarete derived fields * aarete derived prompt template added * get_aarete_derived function added * fixed valid values issue * fixed valid values * modified aarete derived for dataframe * Merge remote-tracking branch 'origin/main' into feature/contract_to_aarete_derived_fields * add commented-out test shell * Fixed one_to_n file running * updated PIPE_FORMAT_INSTRUCTIONS * Fix conflict * Update FieldSet class filters * Updated crosswalk mapping integration * Merged main into feature/contract_to_aarete_derived_fields * Merge branch 'main' into feature/contract_to_aarete_derived_fields * Merged main into feature/contract_to_aarete_derived_fields * Removed test functionality * aareted derived llm values for unique column values * investment_prompts.py edited online with Bitbucket * Removed print statements * Changed load_from_file - field_set = FieldSet(file_path=fields.json, relationship=one-to-one, field_type=True, base_field=True) * Throw type error * Remove ellipsis Approved-by: Katon Minhas
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
# import pytest
|
|
# from unittest.mock import patch, MagicMock
|
|
# from src.investment.aarete_derived import get_aarete_derived
|
|
|
|
# @pytest.fixture
|
|
# def mock_config():
|
|
# with patch('src.investment.aarete_derived.config') as mock_config:
|
|
# mock_config.FIELD_JSON_PATH = 'mock_path'
|
|
# mock_config.MODEL_ID_CLAUDE35_SONNET = 'mock_model_id'
|
|
# yield mock_config
|
|
|
|
# @pytest.fixture
|
|
# def mock_field_set():
|
|
# mock_field = MagicMock()
|
|
# mock_field.field_name = 'derived_field'
|
|
# mock_field.base_field = 'base_field'
|
|
# mock_field.valid_values = 'valid_values_list'
|
|
|
|
# mock_field_set = MagicMock()
|
|
# mock_field_set.fields = [mock_field]
|
|
|
|
# with patch('src.investment.aarete_derived.FieldSet', return_value=mock_field_set):
|
|
# yield mock_field_set
|
|
|
|
# @pytest.fixture
|
|
# def mock_investment_values():
|
|
# with patch('src.investment.aarete_derived.investment_values') as mock_investment_values:
|
|
# mock_investment_values.valid_values_list = ['valid_value1', 'valid_value2']
|
|
# yield mock_investment_values
|
|
|
|
# @pytest.fixture
|
|
# def mock_llm_utils():
|
|
# with patch('src.investment.aarete_derived.llm_utils') as mock_llm_utils:
|
|
# mock_llm_utils.invoke_claude.return_value = '|extracted_value|'
|
|
# yield mock_llm_utils
|
|
|
|
# @pytest.fixture
|
|
# def mock_string_utils():
|
|
# with patch('src.investment.aarete_derived.string_utils') as mock_string_utils:
|
|
# mock_string_utils.extract_text_from_delimiters.return_value = 'extracted_value'
|
|
# yield mock_string_utils
|
|
|
|
# def test_get_aarete_derived(mock_config, mock_field_set, mock_investment_values, mock_llm_utils, mock_string_utils):
|
|
# contract_answers = {'base_field': 'base_value'}
|
|
# filename = 'test_file.txt'
|
|
|
|
# result = get_aarete_derived(contract_answers, filename)
|
|
|
|
# assert result['derived_field'] == 'extracted_value'
|
|
# mock_llm_utils.invoke_claude.assert_called_once()
|
|
# mock_string_utils.extract_text_from_delimiters.assert_called_once_with('|extracted_value|', mock_field_set.fields[0].Delimiter.PIPE) |