Import test code in run_tests.py

Change-Id: Ib0a1235cbe1039261e8f1a0425000e131a868f46
Merge "Import test code in run_tests.py"