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