Skip to content
This repository has been archived by the owner on Jan 15, 2024. It is now read-only.

Latest commit

 

History

History

tests

Folders and files

NameName
Last commit message
Last commit date

parent directory

..

data_cli

data_cli

 
 

process_cli

process_cli

 
 

torch

torch

 
 

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_mt5.py

test_models_mt5.py

 
 

test_models_roberta.py

test_models_roberta.py

 
 

test_models_t5.py

test_models_t5.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_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 --forked --device="cpu" .

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

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

To test only for gpu device, use the following command

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

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

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

In addition, to run all the tests, you should add the --runslow flag

python3 -m pytest --forked --device="gpu" --runslow test_models.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.

Also, we include the scheduled testing scripts for nlp_process in process_cli, and 'nlp_data' in data_cli.