wehub-resource-sync
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
|
|
""" Tests core library functions. By default, runs on MongoDB - to change:
|
|
|
|
`LLMWareConfig().set_active_db("sqlite")
|
|
|
|
"""
|
|
|
|
|
|
import os
|
|
import tempfile
|
|
from llmware.configs import LLMWareConfig
|
|
from llmware.library import Library, LibraryCatalog
|
|
from llmware.setup import Setup
|
|
|
|
|
|
def test_core_library_functions():
|
|
|
|
# Library creations
|
|
library_name = "LibraryABC"
|
|
library = Library().create_new_library(library_name)
|
|
|
|
# Get all library names:
|
|
all_library_names = []
|
|
for library_card in LibraryCatalog().all_library_cards():
|
|
all_library_names.append(library_card["library_name"])
|
|
|
|
# Ensure library name exists
|
|
assert library_name in all_library_names
|
|
|
|
# Add a few files
|
|
sample_files_path = Setup().load_sample_files()
|
|
library.add_files(os.path.join(sample_files_path,"SmallLibrary"))
|
|
|
|
# Library Export
|
|
temp_export_dir = tempfile.gettempdir()
|
|
json_export_path = library.export_library_to_jsonl_file(temp_export_dir, "lib_export")
|
|
assert os.path.getsize(json_export_path) > 100
|
|
os.remove(json_export_path)
|
|
|
|
# dump whole parsed library with only text
|
|
text_export_path = library.export_library_to_txt_file(temp_export_dir, "lib_export")
|
|
assert os.path.getsize(text_export_path) > 100
|
|
os.remove(text_export_path)
|
|
|
|
# Delete the library
|
|
library.delete_library(confirm_delete=True)
|
|
|
|
# Get all library names:
|
|
all_library_names = []
|
|
for library_card in LibraryCatalog().all_library_cards():
|
|
all_library_names.append(library_card["library_name"])
|
|
|
|
# Ensure library name does not exist
|
|
assert library_name not in all_library_names
|
|
|
|
|