diff options
Diffstat (limited to 'absl/testing/tests/absltest_fail_fast_test_helper.py')
-rw-r--r-- | absl/testing/tests/absltest_fail_fast_test_helper.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/absl/testing/tests/absltest_fail_fast_test_helper.py b/absl/testing/tests/absltest_fail_fast_test_helper.py index c684d12..dbef068 100644 --- a/absl/testing/tests/absltest_fail_fast_test_helper.py +++ b/absl/testing/tests/absltest_fail_fast_test_helper.py @@ -18,6 +18,7 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +import os import sys from absl.testing import absltest @@ -44,4 +45,7 @@ class ClassA(absltest.TestCase): if __name__ == '__main__': - absltest.main() + if os.getenv('USE_ARGV', '0') == '1': + absltest.main(argv=sys.argv) + else: + absltest.main() |