def test_select(self):
fridge_meter = ElecMeter()
fridge = Appliance({'type':'fridge', 'instance':1})
fridge_meter.appliances = [fridge]
mg = MeterGroup([fridge_meter])
self.assertEqual(mg.select_using_appliances(category='cold'), mg)
def test_select(self):
fridge_meter = ElecMeter()
fridge = Appliance({'type': 'fridge', 'instance': 1})
fridge_meter.appliances = [fridge]
mg = MeterGroup([fridge_meter])
self.assertEqual(mg.select_using_appliances(category='cold'), mg)
def test_getitem(self):
fridge_meter = ElecMeter()
fridge = Appliance({'type': 'fridge', 'instance': 1})
fridge_meter.appliances = [fridge]
mg = MeterGroup([fridge_meter])
# test good keys
for key in [
'fridge', ('fridge', 1), {
'type': 'fridge'
}, {
'type': 'fridge',
'instance': 1
}
]:
self.assertEqual(mg[key], fridge_meter)
# test bad key values
for key in [
'foo', ('foo', 2), ('fridge', 2), {
'type': 'fridge',
'instance': -12
}
]:
with self.assertRaises(KeyError):
mg[key]
# test bad key types
for key in [True, False, ['fridge']]:
with self.assertRaises(TypeError):
mg[key]
def test_getitem(self):
fridge_meter = ElecMeter()
fridge = Appliance({'type':'fridge', 'instance':1})
fridge_meter.appliances = [fridge]
mg = MeterGroup([fridge_meter])
# test good keys
for key in ['fridge', ('fridge', 1), {'type':'fridge'},
{'type':'fridge', 'instance': 1}]:
self.assertEqual(mg[key], fridge_meter)
# test bad key values
for key in ['foo', ('foo', 2), ('fridge', 2),
{'type':'fridge', 'instance': -12}]:
with self.assertRaises(KeyError):
mg[key]
# test bad key types
for key in [True, False, (1,2,3), ['fridge']]:
with self.assertRaises(TypeError):
mg[key]
