python - 在 TensorFlow 中调试 python 测试

Question

我们想在 TensorFlow 中调试 Python 测试，例如 sparse_split_op_test 和 string_to_hash_bucket_op_test

我们可以使用 gdb 调试其他 c++ 测试，但是我们找不到调试 python 测试的方法。

有没有一种方法可以调试通过 Bazel 测试命令运行的特定 python 测试用例(例如，bazel test//tensorflow/python/kernel_tests:sparse_split_op_test)

Answer 1

我会先构建测试:

bazel build //tensorflow/python/kernel_tests:sparse_split_op_test

然后在生成的 Python 二进制文件上使用 pdb:

pdb bazel-bin/tensorflow/python/kernel_tests/sparse_split_op_test

这似乎对我逐步完成测试的前几行有用。