17db96d56Sopenharmony_ci# Set of tests run by default if --pgo is specified.  The tests below were
27db96d56Sopenharmony_ci# chosen based on the following criteria: either they exercise a commonly used
37db96d56Sopenharmony_ci# C extension module or type, or they run some relatively typical Python code.
47db96d56Sopenharmony_ci# Long running tests should be avoided because the PGO instrumented executable
57db96d56Sopenharmony_ci# runs slowly.
67db96d56Sopenharmony_ciPGO_TESTS = [
77db96d56Sopenharmony_ci    'test_array',
87db96d56Sopenharmony_ci    'test_base64',
97db96d56Sopenharmony_ci    'test_binascii',
107db96d56Sopenharmony_ci    'test_binop',
117db96d56Sopenharmony_ci    'test_bisect',
127db96d56Sopenharmony_ci    'test_bytes',
137db96d56Sopenharmony_ci    'test_bz2',
147db96d56Sopenharmony_ci    'test_cmath',
157db96d56Sopenharmony_ci    'test_codecs',
167db96d56Sopenharmony_ci    'test_collections',
177db96d56Sopenharmony_ci    'test_complex',
187db96d56Sopenharmony_ci    'test_dataclasses',
197db96d56Sopenharmony_ci    'test_datetime',
207db96d56Sopenharmony_ci    'test_decimal',
217db96d56Sopenharmony_ci    'test_difflib',
227db96d56Sopenharmony_ci    'test_embed',
237db96d56Sopenharmony_ci    'test_float',
247db96d56Sopenharmony_ci    'test_fstring',
257db96d56Sopenharmony_ci    'test_functools',
267db96d56Sopenharmony_ci    'test_generators',
277db96d56Sopenharmony_ci    'test_hashlib',
287db96d56Sopenharmony_ci    'test_heapq',
297db96d56Sopenharmony_ci    'test_int',
307db96d56Sopenharmony_ci    'test_itertools',
317db96d56Sopenharmony_ci    'test_json',
327db96d56Sopenharmony_ci    'test_long',
337db96d56Sopenharmony_ci    'test_lzma',
347db96d56Sopenharmony_ci    'test_math',
357db96d56Sopenharmony_ci    'test_memoryview',
367db96d56Sopenharmony_ci    'test_operator',
377db96d56Sopenharmony_ci    'test_ordered_dict',
387db96d56Sopenharmony_ci    'test_patma',
397db96d56Sopenharmony_ci    'test_pickle',
407db96d56Sopenharmony_ci    'test_pprint',
417db96d56Sopenharmony_ci    'test_re',
427db96d56Sopenharmony_ci    'test_set',
437db96d56Sopenharmony_ci    'test_sqlite3',
447db96d56Sopenharmony_ci    'test_statistics',
457db96d56Sopenharmony_ci    'test_struct',
467db96d56Sopenharmony_ci    'test_tabnanny',
477db96d56Sopenharmony_ci    'test_time',
487db96d56Sopenharmony_ci    'test_unicode',
497db96d56Sopenharmony_ci    'test_xml_etree',
507db96d56Sopenharmony_ci    'test_xml_etree_c',
517db96d56Sopenharmony_ci]
527db96d56Sopenharmony_ci
537db96d56Sopenharmony_cidef setup_pgo_tests(ns):
547db96d56Sopenharmony_ci    if not ns.args and not ns.pgo_extended:
557db96d56Sopenharmony_ci        # run default set of tests for PGO training
567db96d56Sopenharmony_ci        ns.args = PGO_TESTS[:]
57