From 4ad50b8d2b82b40b17ecf692495ef63e92a13bd2 Mon Sep 17 00:00:00 2001
From: Richard Belleville <rbellevi@google.com>
Date: Wed, 23 Oct 2019 10:34:59 -0700
Subject: [PATCH] Don't run benchmark in test runner

---
 .../grpcio_tests/tests/stress/unary_stream_benchmark.py   | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py b/src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py
index 6108808c87b..f8d48be17ad 100644
--- a/src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py
+++ b/src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py
@@ -6,8 +6,12 @@ import sys
 import time
 import contextlib
 
-from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2
-from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2_grpc
+try:
+    from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2
+    from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2_grpc
+except ImportError:
+    sys.stderr.write("This test must be run from Bazel.\n")
+    sys.exit(1)
 
 _PORT = 5741
 _MESSAGE_SIZE = 4