class_docstr: Test InMemoryDocumentStore's specific features
class_docstr_tok: ["Test","InMemoryDocumentStore","'s","specific","features"]
code: def test_embedding_retrieval(self):
      docstore = InMemoryDocumentStore(
          embedding_similarity_function="cosine"
      )
      # Tests if the embedding retrieval method returns the correct document based on the input query embedding.
      docs = [
          Document(
              content="Hello world", embedding=[0.1, 0.2, 0.3, 0.4]
          ),
          Document(
              content="Haystack supports multiple languages",
              embedding=[1.0, 1.0, 1.0, 1.0],
          ),
      ]
      docstore.write_documents(docs)
      results = docstore.embedding_retrieval(
          query_embedding=[0.1, 0.1, 0.1, 0.1],
          top_k=1,
          filters={},
          scale_score=False,
      )
      assert len(results) == 1
      assert (
          results[0].content == "Haystack supports multiple languages"
      )
  
imports: import logging
  from unittest.mock import patch
  import pandas
  import pytest
  from haystack import Document
  from haystack.document_stores.errors import DocumentStoreError, DuplicateDocumentError
  from haystack.document_stores.in_memory import InMemoryDocumentStore
  from haystack.testing.document_store import DocumentStoreBaseTests
label_desc: Use image node_id 20 for calling the TestMemoryDocumentStore obj's underlying member method code with example usage: obj.test_embedding_retrieval() without return types