def mock_tmy3_weather_source():
tmp_url = "sqlite:///{}/weather_cache.db".format(tempfile.mkdtemp())
ws = TMY3WeatherSource('724838', tmp_url, preload=False)
ws.client = MockWeatherClient()
ws._load_data()
return ws
def mock_isd_weather_source():
tmp_url = "sqlite:///{}/weather_cache.db".format(tempfile.mkdtemp())
ws = ISDWeatherSource('722880', tmp_url)
ws.client = MockWeatherClient()
return ws
def mock_tmy3_weather_source():
tmp_dir = tempfile.mkdtemp()
ws = TMY3WeatherSource("724838", tmp_dir, preload=False)
ws.client = MockWeatherClient()
ws._load_data()
return ws
def mock_isd_weather_source():
tmp_dir = tempfile.mkdtemp()
ws = ISDWeatherSource("722880", tmp_dir)
ws.client = MockWeatherClient()
return ws
def setUpTestData(cls):
cls.user = User.objects.create_user('john', '*****@*****.**',
'johnpassword')
cls.project = create_project(
spec={
"project_id":
"ABCD",
"project_owner":
cls.user.projectowner,
"baseline_period_end":
datetime(2012, 1, 1, tzinfo=pytz.UTC),
"reporting_period_start":
datetime(2012, 2, 1, tzinfo=pytz.UTC),
"zipcode":
"91104",
"traces": [
{
"interpretation": "NG_C_S",
"unit": "THM",
"start": "2010-01-01",
"end": "2014-12-31",
"freq": "MS",
"value": 1,
"nans": set(range(0, 60, 20)),
"estimated": set(range(3, 60, 15)),
},
{
"interpretation": "NG_C_S",
"unit": "THM",
"start": "2011-09-01",
"end": "2014-12-31",
"freq": "D",
"value": 2,
"nans": set(range(0, 1000, 20)),
"estimated": set(range(3, 1000, 15)),
},
{
"interpretation": "E_C_S",
"unit": "KWH",
"start": "2011-01-01",
"end": "2014-12-31",
"freq": "15T",
"value": 0.04,
"nans": set(range(0, 96 * 365 * 4, 200)),
"estimated": set(range(3, 96 * 365 * 4, 150)),
},
{
"interpretation": "E_C_S",
"unit": "KWH",
"start": "2011-01-01",
"end": "2014-12-31",
"freq": "H",
"value": 0.4,
"nans": set(range(0, 96 * 365 * 4, 200)),
"estimated": set(range(3, 96 * 365 * 4, 150)),
},
{
"interpretation": "E_OSG_U",
"unit": "KWH",
"start": "2012-01-15",
"end": "2014-12-31",
"freq": "H",
"value": 0.3,
"nans": set(range(0, 96 * 365 * 4, 200)),
"estimated": set(range(3, 96 * 365 * 4, 150)),
},
{
"interpretation": "E_OSG_U",
"unit": "KWH",
"start": "2010-01-01",
"end": "2014-12-31",
"freq": "30T",
"value": 0.1,
"nans": set(range(0, 96 * 365 * 4, 200)),
"estimated": set(range(3, 96 * 365 * 4, 150)),
},
],
})
cls.project.run_meter()
tmp_dir = tempfile.mkdtemp()
wns = TMY3WeatherSource("724838", tmp_dir, preload=False)
wns.client = MockWeatherClient()
wns._load_data()
cls.weather_normal_source = wns
tmp_dir = tempfile.mkdtemp()
ws = ISDWeatherSource("722880", tmp_dir)
ws.client = MockWeatherClient()
cls.weather_source = ws
