Index: build/android/pylib/instrumentation/instrumentation_test_instance.py |
diff --git a/build/android/pylib/instrumentation/instrumentation_test_instance.py b/build/android/pylib/instrumentation/instrumentation_test_instance.py |
index f3c3d3c0891e4b002f9538774a849b1b5b8c1825..278ffb2f7212efa6c25b77ebacf39a202db183d6 100644 |
--- a/build/android/pylib/instrumentation/instrumentation_test_instance.py |
+++ b/build/android/pylib/instrumentation/instrumentation_test_instance.py |
@@ -163,12 +163,12 @@ def ParseCommandLineFlagParameters(annotations): |
""" |
ParamsTuple = collections.namedtuple('ParamsTuple', ['add', 'remove']) |
parameterized_tests = [] |
- if _PARAMETERIZED_TEST_ANNOTATION in annotations: |
- parameterized_tests = [annotations[_PARAMETERIZED_TEST_ANNOTATION]] |
- elif _PARAMETERIZED_TEST_SET_ANNOTATION in annotations: |
+ if _PARAMETERIZED_TEST_SET_ANNOTATION in annotations: |
if annotations[_PARAMETERIZED_TEST_SET_ANNOTATION]: |
parameterized_tests = annotations[ |
_PARAMETERIZED_TEST_SET_ANNOTATION].get('tests', []) |
+ elif _PARAMETERIZED_TEST_ANNOTATION in annotations: |
+ parameterized_tests = [annotations[_PARAMETERIZED_TEST_ANNOTATION]] |
else: |
return None |