def setup_function(): inject.reinject_decorated_tables() inject.remove_injectable('skim_dict') inject.remove_injectable('skim_stack') configs_dir = os.path.join(os.path.dirname(__file__), 'configs') inject.add_injectable("configs_dir", configs_dir) output_dir = os.path.join(os.path.dirname(__file__), 'output') inject.add_injectable("output_dir", output_dir) data_dir = os.path.join(os.path.dirname(__file__), 'data') inject.add_injectable("data_dir", data_dir) inject.clear_cache() tracing.config_logger()
def setup_function(): inject.reinject_decorated_tables() configs_dir = os.path.join(os.path.dirname(__file__), 'configs') inject.add_injectable("configs_dir", configs_dir) output_dir = os.path.join(os.path.dirname(__file__), 'output') inject.add_injectable("output_dir", output_dir) data_dir = os.path.join(os.path.dirname(__file__), 'data') inject.add_injectable("data_dir", data_dir) inject.clear_cache() tracing.config_logger() tracing.delete_output_files('csv') tracing.delete_output_files('txt') tracing.delete_output_files('yaml')
def test_load_cached_accessibility(): inject.clear_cache() inject.reinject_decorated_tables() data_dir = [ os.path.join(os.path.dirname(__file__), 'data'), example_path('data') ] setup_dirs(data_dir=data_dir) # # add OPTIONAL ceched table accessibility to input_table_list # activitysim.abm.tables.land_use.accessibility() will load this table if listed here # presumably independently calculated outside activitysim or a cached copy created during a previous run # settings = config.read_settings_file('settings.yaml', mandatory=True) input_table_list = settings.get('input_table_list') input_table_list.append({ 'tablename': 'accessibility', 'filename': 'cached_accessibility.csv', 'index_col': 'zone_id' }) inject_settings(households_sample_size=HOUSEHOLDS_SAMPLE_SIZE, input_table_list=input_table_list) _MODELS = [ 'initialize_landuse', # 'compute_accessibility', # we load accessibility table ordinarily created by compute_accessibility 'initialize_households', ] pipeline.run(models=_MODELS, resume_after=None) accessibility_df = pipeline.get_table("accessibility") assert 'auPkRetail' in accessibility_df pipeline.close_pipeline() inject.clear_cache() close_handlers()
def teardown_function(func): inject.clear_cache() inject.reinject_decorated_tables()
def teardown_function(func): inject.clear_cache() inject.reinject_decorated_tables()
def teardown_function(func): orca.clear_all() orca.enable_cache() # be nice to the others tests that expect decorated injectables to be there inject.reinject_decorated_tables()