def setUp(self): # set up expected values from base client bc = BaseClient() self.MARKET_CHOICES = bc.MARKET_CHOICES self.FREQUENCY_CHOICES = bc.FREQUENCY_CHOICES # set up other expected values self.BA_CHOICES = BALANCING_AUTHORITIES.keys()
def setUp(self): super(TestSVERIGenMix, self).setUp() self.bas = [ k for k, v in BALANCING_AUTHORITIES.items() if v['module'] == 'sveri' ]
def setUp(self): super(TestSVERILoad, self).setUp() self.bas = [k for k, v in BALANCING_AUTHORITIES.items() if v['module'] == 'sveri']
'coal': 'coal', 'hydro': 'hydro', 'natgas': 'gas', 'nuclear': 'nuclear', 'oil': 'oil', 'refuse': 'biomass', 'solar': 'solar', 'wind': 'wind', 'other': 'unknown' } # Note: the list of ISOs is taken from https://github.com/WattTime/pyiso/blob/master/pyiso/__init__.py#L20 # in order to instantiate every possible client ISO_LIST = {} # Get a unique list of balancing authorities such that each (module, client) pair is unique for k, v in BALANCING_AUTHORITIES.iteritems(): # We obivously exclude the EU if k == 'EU': continue # Those clients are not working: if v['class'] in ['NVEnergyClient', 'SPPClient']: continue ISO_LIST[v['class'] + v['module']] = k ISO_LIST = ISO_LIST.values() def fetch_production(country_code='US', session=None): obj = { 'countryCode': country_code, 'production': {}, 'source': 'github.com/WattTime/pyiso' } for iso in ISO_LIST: cli = client_factory(iso)
def setUp(self): super(TestSVERIGenMix, self).setUp() self.bas = [k for k, v in BALANCING_AUTHORITIES.items() if v["module"] == "sveri"]
def get_authorities() -> typing.List[Authority]: """Get a list of the balancing authorities.""" return [Authority(code=code) for code in BALANCING_AUTHORITIES.keys()]