From 85f2cdbee762e5dd10ddfd8b41498f5aa552341c Mon Sep 17 00:00:00 2001 From: uael Date: Tue, 14 Mar 2023 21:56:57 +0000 Subject: [PATCH] Tests: update `suite_name.py` to latest mobly Bug: 270963463 Test: atest NearbyMultiDevicesTestSuite Change-Id: Ia523c51c6c834534a40478dcae147a7de6e12f0e --- nearby/tests/multidevices/host/suite_main.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/nearby/tests/multidevices/host/suite_main.py b/nearby/tests/multidevices/host/suite_main.py index 4f5d48cc90..9a580fb0b5 100644 --- a/nearby/tests/multidevices/host/suite_main.py +++ b/nearby/tests/multidevices/host/suite_main.py @@ -31,11 +31,9 @@ _TEST_CLASSES_LIST = [ ] -def _valid_argument(arg: str) -> bool: - return arg.startswith(('--config', '-c', '--tests', '--test_case')) - - if __name__ == '__main__': logging.basicConfig(filename=_BOOTSTRAP_LOGGING_FILENAME, level=logging.INFO) - suite_runner.run_suite(argv=[arg for arg in sys.argv if _valid_argument(arg)], - test_classes=_TEST_CLASSES_LIST) + if '--' in sys.argv: + index = sys.argv.index('--') + sys.argv = sys.argv[:1] + sys.argv[index + 1:] + suite_runner.run_suite(test_classes=_TEST_CLASSES_LIST)