Fix benchmark_device

If user has not set its environment there are no instructions on how
to fix it.

Test: NA
Bug: 310747667
Change-Id: I43ab2416991399ea6383a88b384532a9dba2894c

1 files changed, 6 insertions, 0 deletions

diff --git a/benchmark_device.py b/benchmark_device.py
index 4d0cf49a..4a543480 100755
--- a/benchmark_device.py
+++ b/benchmark_device.py
@@ -18,9 +18,15 @@
 import os
 import statistics
 import subprocess
+import sys
 import tempfile
 import time
 
+# Make sure environment is setup, otherwise "adb" module is not available.
+if os.getenv("ANDROID_BUILD_TOP") is None:
+    print("Run source/lunch before running " + sys.argv[0])
+    sys.exit()
+
 import adb
 
 def lock_min(device):