:mod:`biotransformers.tests.test_embeddings`
============================================

.. py:module:: biotransformers.tests.test_embeddings

.. autoapi-nested-parse::

   Test module for testing embeddings function



Module Contents
---------------


Functions
~~~~~~~~~

.. autoapisummary::

   biotransformers.tests.test_embeddings.test_embeddings_type_and_shape
   biotransformers.tests.test_embeddings.test_embeddings_type_and_shape_fasta



Attributes
~~~~~~~~~~

.. autoapisummary::

   biotransformers.tests.test_embeddings.test_params


.. data:: test_params
   :annotation: = [[1, ['cls', 'mean']], [2, ['full', 'mean', 'cls']], [10, ['cls', 'full']]]

   

.. function:: test_embeddings_type_and_shape(init_model, sequences, batch_size, pool_mode)


.. function:: test_embeddings_type_and_shape_fasta(init_model, fasta_path, lengths_sequence_fasta, batch_size, pool_mode)