Merged in refactor/prompt_refactor (pull request #353)
READY: Refactor/prompt refactor * Prompt refactor * refactor imports * Prompt refactor * Import refactor * Remove excess prompts from investment_prompts.py Approved-by: Alex Galarce
This commit is contained in:
@@ -1,7 +1,8 @@
|
||||
import pytest
|
||||
from src.prompts import preprocessing_prompts
|
||||
import src.utils as utils
|
||||
from src.utils.table_utils import align_and_format_tables
|
||||
from src import (prompts, config)
|
||||
from src import (config)
|
||||
import copy
|
||||
|
||||
class TestAlignAndFormatTables:
|
||||
@@ -30,7 +31,7 @@ class TestAlignAndFormatTables:
|
||||
}
|
||||
filename = "test_file.txt"
|
||||
|
||||
expected_prompt = prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
|
||||
expected_prompt = preprocessing_prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
|
||||
input_text_dict = copy.deepcopy(text_dict)
|
||||
|
||||
result = align_and_format_tables(text_dict, filename)
|
||||
@@ -91,7 +92,7 @@ class TestAlignAndFormatTables:
|
||||
}
|
||||
filename = "test_file.txt"
|
||||
|
||||
expected_prompt = prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
|
||||
expected_prompt = preprocessing_prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
|
||||
input_text_dict = copy.deepcopy(text_dict)
|
||||
|
||||
result = align_and_format_tables(text_dict, filename)
|
||||
|
||||
Reference in New Issue
Block a user