Skip to content

Commit 645a0c8

Browse files
authored
Merge pull request llmware-ai#1075 from smit23patel/patch-1
Update test_path_overrides.py
2 parents c6d47a9 + d308bfc commit 645a0c8

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

‎tests/configs/test_path_overrides.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
21
""" Tests paths both for a standard account setup and custom account setup.
32
43
Note: by default, this test will run on MongoDB - to switch DB used - add one-line:
@@ -27,7 +26,7 @@ def test_standard_account_setup():
2726
# run test query
2827
results = Query(library).text_query("base salary")
2928

30-
assert len(results) > 0
29+
assert len(results) > 0, "Expected results to be greater than 0 for standard account setup."
3130

3231

3332
# Test overriding llmware_data folder
@@ -46,19 +45,20 @@ def test_setup_custom_location():
4645
if not os.path.exists(LLMWareConfig().get_llmware_path()):
4746
LLMWareConfig.setup_llmware_workspace()
4847

49-
assert new_home_folder in LLMWareConfig.get_home()
50-
assert new_home_folder in LLMWareConfig.get_llmware_path() and new_llmware_path in LLMWareConfig.get_llmware_path()
51-
assert new_home_folder in LLMWareConfig.get_library_path() and new_llmware_path in LLMWareConfig.get_library_path()
48+
assert new_home_folder in LLMWareConfig.get_home(), "Custom home folder not set correctly."
49+
assert new_home_folder in LLMWareConfig.get_llmware_path() and new_llmware_path in LLMWareConfig.get_llmware_path(), "Custom LLMWare path not set correctly."
50+
assert new_home_folder in LLMWareConfig.get_library_path() and new_llmware_path in LLMWareConfig.get_library_path(), "Library path not set correctly."
5251

5352
# test with library setup and adding files
5453
library_name = "library_with_custom_path"
5554
library = Library().create_new_library(library_name)
5655
sample_files_path = Setup().load_sample_files()
5756
library.add_files(os.path.join(sample_files_path,"Agreements"))
5857

59-
assert new_home_folder in library.library_main_path
58+
assert new_home_folder in library.library_main_path, "Custom path not found in library main path."
6059

6160
results = Query(library).text_query("salary")
6261

6362
assert len(results) > 0
6463

64+

0 commit comments

Comments
 (0)