:mod:`biotransformers.tests.test_logits`
========================================

.. py:module:: biotransformers.tests.test_logits

.. autoapi-nested-parse::

   Test module for testing logits function



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


Functions
~~~~~~~~~

.. autoapisummary::

   biotransformers.tests.test_logits.test_logits_type
   biotransformers.tests.test_logits.test_logits_type_fasta



Attributes
~~~~~~~~~~

.. autoapisummary::

   biotransformers.tests.test_logits.test_params
   biotransformers.tests.test_logits.test_params_fasta


.. data:: test_params
   :annotation: = [[1, 'forward'], [2, 'masked'], [10, 'forward']]

   

.. data:: test_params_fasta
   :annotation: = [[2, 'forward']]

   

.. function:: test_logits_type(init_model, batch_size, sequences, pass_mode)


.. function:: test_logits_type_fasta(init_model, batch_size, fasta_path, lengths_sequence_fasta, pass_mode)