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,