def setup_class(cls): # initialize with data to avoid pulling multiple times cls.sim_config = Config.make_sim_config( identifier=[ os.environ.get( "TEST_FLATFILES_GBQ_IDENTIFIER"), # has all holds "9999999", # file not found ], latitude=33.481136, longitude=-112.078232, start_utc="2019-01-02 00:00:00", end_utc="2019-02-01 00:00:00", min_sim_period="3D", sim_step_size_seconds=300, output_step_size_seconds=300, ) cls.data_clients = [] # set local_cache=None to test connection with GCS cls.data_client = DataClient( source=GBQDataSource( gcp_project=os.environ.get("FLATFILE_GOOGLE_CLOUD_PROJECT"), gbq_table=os.environ.get("FLATFILES_GBQ_TABLE"), data_spec=FlatFilesSpec(), local_cache=None, ), destination=LocalDestination( local_cache=os.environ.get("LOCAL_CACHE_DIR"), data_spec=FlatFilesSpec(), ), nrel_dev_api_key=os.environ.get("NREL_DEV_API_KEY"), nrel_dev_email=os.environ.get("NREL_DEV_EMAIL"), archive_tmy3_dir=os.environ.get("ARCHIVE_TMY3_DIR"), archive_tmy3_meta=os.environ.get("ARCHIVE_TMY3_META"), archive_tmy3_data_dir=os.environ.get("ARCHIVE_TMY3_DATA_DIR"), ep_tmy3_cache_dir=os.environ.get("EP_TMY3_CACHE_DIR"), simulation_epw_dir=os.environ.get("SIMULATION_EPW_DIR"), ) for _idx, _sim_config in cls.sim_config.iterrows(): dc = copy.deepcopy(cls.data_client) dc.sim_config = _sim_config if _sim_config["identifier"] == "9999999": with pytest.raises(ValueError): dc.get_data() else: dc.get_data() cls.data_clients.append(dc)
"latitude": 51.217373, "longitude": -114.296019, "start_utc": "2019-03-09", "end_utc": "2019-03-15", "min_sim_period": "1D", "sim_step_size_seconds": 60, "output_step_size_seconds": 300, }, "data_client": { "is_local_source": False, "is_gcs_source": False, "is_gbq_source": True, "gcp_project": os.environ.get("DYD_GOOGLE_CLOUD_PROJECT"), "gcs_uri_base": None, "gbq_table": os.environ.get("FLATFILES_GBQ_TABLE"), "source_data_spec": FlatFilesSpec(), "source_local_cache": os.environ.get("LOCAL_CACHE_DIR"), "is_local_destination": True, "is_gcs_destination": False, "is_gbq_destination": False, "destination_data_spec": FlatFilesSpec(), "destination_local_cache": os.environ.get("LOCAL_CACHE_DIR"), }, "building_model": { "is_energyplus_building": True, "idf_name": "heatedbsmt_2story_2300sqft_gasfurnace_AC.idf", "epw_name": "USA_IL_Chicago-OHare.Intl.AP.725300_TMY3.epw", "building_config": { "infiltration_ventilation": { "ach50": 10, "wsf": 0.6,