from llama_index.core.storage.kvstore.types import BaseKVStore
from llama_index.storage.docstore.azurecosmosnosql import AzureCosmosNoSqlDocumentStore


def test_class():
    names_of_base_classes = [b.__name__ for b in AzureCosmosNoSqlDocumentStore.__mro__]
    assert BaseKVStore.__name__ in names_of_base_classes

from llama_index.core.storage.docstore.keyval_docstore import KVDocumentStore
from llama_index.storage.docstore.azurecosmosnosql import AzureCosmosNoSqlDocumentStore


def test_class():
    names_of_base_classes = [b.__name__ for b in AzureCosmosNoSqlDocumentStore.__mro__]
    assert KVDocumentStore.__name__ in names_of_base_classes
