diff --git a/libs/databricks/tests/integration_tests/test_embeddings.py b/libs/databricks/tests/integration_tests/test_embeddings.py
new file mode 100644
index 0000000..d0c6b10
--- /dev/null
+++ b/libs/databricks/tests/integration_tests/test_embeddings.py
@@ -0,0 +1,17 @@
+from langchain_databricks import DatabricksEmbeddings
+
+_TEST_ENDPOINT = "databricks-bge-large-en"
+
+def test_embedding_documents() -> None:
+    documents = ["foo bar"]
+    embedding = DatabricksEmbeddings(endpoint=_TEST_ENDPOINT)
+    output = embedding.embed_documents(documents)
+    assert len(output) == 1
+    assert len(output[0]) > 0
+
+
+def test_embedding_query() -> None:
+    document = "foo bar"
+    embedding = DatabricksEmbeddings(endpoint=_TEST_ENDPOINT)
+    output = embedding.embed_query(document)
+    assert len(output) > 0