Exécuter des tests unitaires Tensorflow
Existe-t-il un moyen d'exécuter les tests unitaires Tensorflow manuellement? Je souhaite effectuer des vérifications d'intégrité tout en modifiant le code source de TF.
Je vois qu'il existe de nombreux fichiers _test.py avec des classes qui effectuent de nombreuses opérations de test et je ne sais pas comment les exécuter. Il devrait y avoir un moyen simple?
La façon la plus simple d'exécuter les tests unitaires TensorFlow est d'utiliser Bazel, en supposant que vous avez téléchargé la source depuis Git :
# All tests (for C++ changes).
$ bazel test //tensorflow/...
# All Python tests (for Python front-end changes).
$ bazel test //tensorflow/python/...
# All tests (with GPU support).
$ bazel test -c opt --config=cuda //tensorflow/...
$ bazel test -c opt --config=cuda //tensorflow/python/...
En plus de la réponse ci-dessus, vous pouvez exécuter des tests individuels comme indiqué au lieu de packages complets, ce qui peut gagner beaucoup de temps.
bazel run //tensorflow/python/kernel_tests:string_split_op_test
bazel run //tensorflow/python:special_math_ops_test
Ou vous pouvez aller dans le répertoire individuel et y exécuter tous les tests
cd python/kernel_tests
bazel run :one_hot_op_test