Check environment by verifying other dirs' existence
Signed-off-by: Yuto Takano <yuto.takano@arm.com>
diff --git a/tests/scripts/check-names.py b/tests/scripts/check-names.py
index e3863cf..b9e028b 100755
--- a/tests/scripts/check-names.py
+++ b/tests/scripts/check-names.py
@@ -162,11 +162,10 @@
Check that the current working directory is the project root, and throw
an exception if not.
"""
- current_dir = os.path.realpath('.')
- root_dir = os.path.dirname(os.path.dirname(
- os.path.dirname(os.path.realpath(__file__))))
- if current_dir != root_dir:
- raise Exception("Must be run from Mbed TLS root")
+ if (not os.path.isdir("include") or
+ not os.path.isdir("tests") or
+ not os.path.isdir("library")):
+ raise Exception("This script must be run from Mbed TLS root")
def get_files(self, extension, directory):
"""
