langchain/libs/partners/astradb/tests/unit_tests/test_imports.py

from langchain_astradb import __all__

EXPECTED_ALL = [
    "AstraDBByteStore",
    "AstraDBStore",
    "AstraDBVectorStore",
]


def test_all_imports() -> None:
    assert sorted(EXPECTED_ALL) == sorted(__all__)
astradb: bootstrapping Astra DB as Partner Package (#16875) Description: This PR introduces a new "Astra DB" Partner Package. So far only the vector store class is _duplicated_ there, all others following once this is validated and established. Along with the move to separate package, incidentally, the class name will change `AstraDB` => `AstraDBVectorStore`. The strategy has been to duplicate the module (with prospected removal from community at LangChain 0.2). Until then, the code will be kept in sync with minimal, known differences (there is a makefile target to automate drift control. Out of convenience with this check, the community package has a class `AstraDBVectorStore` aliased to `AstraDB` at the end of the module). With this PR several bugfixes and improvement come to the vector store, as well as a reshuffling of the doc pages/notebooks (Astra and Cassandra) to align with the move to a separate package. Dependencies: A brand new pyproject.toml in the new package, no changes otherwise. Twitter handle: `@rsprrs` --------- Co-authored-by: Christophe Bornet <cbornet@hotmail.com> Co-authored-by: Erick Friis <erick@langchain.dev> 2024-02-15 23:50:59 +00:00			`from langchain_astradb import __all__`

			`EXPECTED_ALL = [`
astradb[patch]: Add AstraDBStore to langchain-astradb package (#17789) Co-authored-by: Erick Friis <erick@langchain.dev> 2024-02-21 00:54:35 +00:00			`"AstraDBByteStore",`
			`"AstraDBStore",`
astradb: bootstrapping Astra DB as Partner Package (#16875) Description: This PR introduces a new "Astra DB" Partner Package. So far only the vector store class is _duplicated_ there, all others following once this is validated and established. Along with the move to separate package, incidentally, the class name will change `AstraDB` => `AstraDBVectorStore`. The strategy has been to duplicate the module (with prospected removal from community at LangChain 0.2). Until then, the code will be kept in sync with minimal, known differences (there is a makefile target to automate drift control. Out of convenience with this check, the community package has a class `AstraDBVectorStore` aliased to `AstraDB` at the end of the module). With this PR several bugfixes and improvement come to the vector store, as well as a reshuffling of the doc pages/notebooks (Astra and Cassandra) to align with the move to a separate package. Dependencies: A brand new pyproject.toml in the new package, no changes otherwise. Twitter handle: `@rsprrs` --------- Co-authored-by: Christophe Bornet <cbornet@hotmail.com> Co-authored-by: Erick Friis <erick@langchain.dev> 2024-02-15 23:50:59 +00:00			`"AstraDBVectorStore",`
			`]`


			`def test_all_imports() -> None:`
			`assert sorted(EXPECTED_ALL) == sorted(__all__)`