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.