Skip to content

Latest commit

 

History

History
 
 

tests

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
test_attention_cell.py
test_attention_cell.py
 
 
test_data_batchify.py
test_data_batchify.py
 
 
test_data_filtering.py
test_data_filtering.py
 
 
test_data_loading.py
test_data_loading.py
 
 
test_data_sampler.py
test_data_sampler.py
 
 
test_data_tokenizers.py
test_data_tokenizers.py
 
 
test_data_vocab.py
test_data_vocab.py
 
 
test_embedding.py
test_embedding.py
 
 
test_gluon_block.py
test_gluon_block.py
 
 
test_initializer.py
test_initializer.py
 
 
test_layers.py
test_layers.py
 
 
test_loss.py
test_loss.py
 
 
test_models.py
test_models.py
 
 
test_models_albert.py
test_models_albert.py
 
 
test_models_bart.py
test_models_bart.py
 
 
test_models_bert.py
test_models_bert.py
 
 
test_models_electra.py
test_models_electra.py
 
 
test_models_gpt2.py
test_models_gpt2.py
 
 
test_models_mobilebert.py
test_models_mobilebert.py
 
 
test_models_roberta.py
test_models_roberta.py
 
 
test_models_transformer.py
test_models_transformer.py
 
 
test_models_transformer_xl.py
test_models_transformer_xl.py
 
 
test_models_xlmr.py
test_models_xlmr.py
 
 
test_op.py
test_op.py
 
 
test_optimizer.py
test_optimizer.py
 
 
test_pytest.py
test_pytest.py
 
 
test_sequence_sampler.py
test_sequence_sampler.py
 
 
test_utils_misc.py
test_utils_misc.py
 
 
test_utils_parameter.py
test_utils_parameter.py
 
 
test_utils_preprocessing.py
test_utils_preprocessing.py
 
 
test_utils_registry.py
test_utils_registry.py
 
 

README.md

Unit Tests

To run the unittests, use the following command

python3 -m pytest --device="cpu" .

To test for certain file, e.g., the test_models_transformer.py, use the following command

python3 -m pytest --device="cpu" test_models_transformer.py

To test only for gpu device, use the following command

python3 -m pytest --device="gpu" test_models_transformer.py

To test both for cpu and gpu device, use the following command

python3 -m pytest --device="cpu" --device="gpu" test_models_transformer.py

Refer to the official guide of pytest for more details.

Naming Convention

The naming convention of the tests are test_{module_name}.py. For example, the test of models/transformer.py will be in test_models_transformer.py. The test of models/init.py is test_models.py.