B   ›\t ã@sœddlZddlZddlZddlmZmZmZddlmZyddl Z dZ Wne k r`dZ YnXdZ e  e e ¡ZeGdd „d eƒƒZed kr˜e ¡dS) éNé)ÚTestCaseÚtemp_directoryÚoverride_env_config)ÚconfigTFz*pyyaml needed for configuration file testscsPeZdZdZ‡fdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z ‡Z S)Ú TestConfigFcstdƒ|_tt|ƒ ¡dS)NZ config_tmp)rÚtmppathÚsuperrÚsetUp)Úself)Ú __class__©ú6lib/python3.7/site-packages/numba/tests/test_config.pyr s zTestConfig.setUpcCstj|jdS)zT Creates a mock launch location. Returns the location path. )Údir)ÚtempfileZmkdtempr)r r r rÚmock_cfg_locationszTestConfig.mock_cfg_locationc Cs:tj |tj¡}t|dƒ}tj||ddWdQRXdS)z< Injects a mock configuration at 'location' ZwtF)Zdefault_flow_styleN)ÚosÚpathÚjoinrZ _config_fnameÚopenÚyamlÚdump)r ÚlocationÚcfgZtmpcfgÚfr r rÚinject_mock_cfg#s zTestConfig.inject_mock_cfgcCs2tƒ}x&ttƒD]}| ¡rtt|ƒ||<qW|S)z8 Gets the current numba config settings )ÚdictrrÚisupperÚgetattr)r ZstoreÚxr r rÚ get_settings+s zTestConfig.get_settingsc Cs|t ¡}| ¡}t |¡tddƒ| ¡}WdQRX| ||¡z"tddƒ| ¡}WdQRXWdt |¡X||fS)z× Returns a config "original" from a location with no config file and then the impact of applying the supplied cfg dictionary as a config file at a location in the returned "current". Ú_N)rÚgetcwdrÚchdirrr r)r rZ original_cwdZ launch_dirZoriginalZcurrentr r rÚcreate_config_effect5s     zTestConfig.create_config_effectcCsˆd}x~tjtjgD]n}| ||ƒdi¡\}}| ||k¡| ||||k¡| ||d¡| |¡| |¡| ||¡qWdS)NZ COLOR_SCHEMEZlight_bg)ÚstrÚupperÚlowerr$Z assertTrueÚ assertEqualÚpop)r ÚkeyZcaseÚorigÚcurrr r rÚ test_configVs  zTestConfig.test_configcCs| i¡\}}| ||¡dS)N)r$r()r r+r,r r rÚtest_empty_configdszTestConfig.test_empty_config) Ú__name__Ú __module__Ú __qualname__Z_numba_parallel_test_r rrr r$r-r.Ú __classcell__r r )r rrs  !rÚ__main__)rrZunittestZsupportrrrZnumbarrZ _HAVE_YAMLÚ ImportErrorZ _skip_msgZskipIfZ needs_yamlrr/Úmainr r r rÚs   X