mirror of https://github.com/arc53/DocsGPT
fix packaging and imports and introduce tests with pytest.
still issues with celery worker.pull/300/head
parent
98a97f34f5
commit
85f9ae5a0a
@ -1,37 +0,0 @@
|
||||
from application.app import get_vectorstore
|
||||
|
||||
|
||||
# Test cases for get_vectorstore function
|
||||
def test_no_active_docs():
|
||||
data = {}
|
||||
assert get_vectorstore(data) == ""
|
||||
|
||||
|
||||
def test_default_active_docs():
|
||||
data = {"active_docs": "default"}
|
||||
assert get_vectorstore(data) == ""
|
||||
|
||||
|
||||
def test_local_default_active_docs():
|
||||
data = {"active_docs": "local/default"}
|
||||
assert get_vectorstore(data) == ""
|
||||
|
||||
|
||||
def test_local_custom_active_docs():
|
||||
data = {"active_docs": "local/custom_index"}
|
||||
assert get_vectorstore(data) == "indexes/local/custom_index"
|
||||
|
||||
|
||||
def test_remote_active_docs():
|
||||
data = {"active_docs": "remote_index"}
|
||||
assert get_vectorstore(data) == "vectors/remote_index"
|
||||
|
||||
|
||||
def test_active_docs_not_in_data():
|
||||
data = {"other_key": "value"}
|
||||
assert get_vectorstore(data) == ""
|
||||
|
||||
|
||||
def test_multiple_slashes_in_active_docs():
|
||||
data = {"active_docs": "local/some/other/index"}
|
||||
assert get_vectorstore(data) == "indexes/local/some/other/index"
|
@ -0,0 +1,28 @@
|
||||
from application.app import get_vectorstore
|
||||
import os
|
||||
|
||||
|
||||
# Test cases for get_vectorstore function
|
||||
def test_no_active_docs():
|
||||
data = {}
|
||||
assert get_vectorstore(data) == os.path.join("application", "")
|
||||
|
||||
|
||||
def test_local_default_active_docs():
|
||||
data = {"active_docs": "local/default"}
|
||||
assert get_vectorstore(data) == os.path.join("application", "")
|
||||
|
||||
|
||||
def test_local_non_default_active_docs():
|
||||
data = {"active_docs": "local/something"}
|
||||
assert get_vectorstore(data) == os.path.join("application", "indexes/local/something")
|
||||
|
||||
|
||||
def test_default_active_docs():
|
||||
data = {"active_docs": "default"}
|
||||
assert get_vectorstore(data) == os.path.join("application", "")
|
||||
|
||||
|
||||
def test_complex_active_docs():
|
||||
data = {"active_docs": "local/other/path"}
|
||||
assert get_vectorstore(data) == os.path.join("application", "indexes/local/other/path")
|
Loading…
Reference in New Issue