def do_failed_load(data, col_name, value):
try:
tmp_data = data.copy()
tmp_data.get(col_name)[1] = value
infile = self.write_file(tmp_data)
s = ConsumeInputSettings()
self.assertFalse(s.load_from_file(infile[1]))
finally:
os.unlink(infile[1])
def testInitial(self):
''' Everything should be empty '''
s = ConsumeInputSettings()
self.assertFalse(s.units)
self.assertFalse(s.burn_type)
self.assertFalse(s.fm_type)
settings = ConsumeInputSettings.NaturalSNames + ConsumeInputSettings.ActivitySNames
for i in settings:
self.assertFalse(s.get(i))
def testSetProperties(self):
''' Exercise setting the 1-value properties '''
s = ConsumeInputSettings()
s.burn_type = 'natural'
self.assertEqual('natural', s.burn_type)
s.burn_type = 'activity'
self.assertEqual('activity', s.burn_type)
s.burn_type = 'kjell'
self.assertEqual('activity', s.burn_type)
s.units = 'kg_ha'
self.assertEqual('kg_ha', s.units)
s.units = 'kjell'
self.assertEqual('kg_ha', s.units)
s.fm_type = 'MEAS-Th'
self.assertEqual('MEAS-Th', s.fm_type)
s.fm_type = 'kjell'
self.assertEqual('MEAS-Th', s.fm_type)
def testDisplaySettings(self):
items = [1, 2, 4]
s = ConsumeInputSettings()
s.burn_type = 'activity'
self.assertTrue(s.set('fuelbeds', items))
fb = s.get('fuelbeds')
self.assertTrue(s.set('slope', [1, 2, 4]))
self.assertTrue(s.set('windspeed', [1, 2, 4]))
self.assertTrue(s.set('days_since_rain', [1, 2, 4]))
self.assertTrue(s.set('fm_10hr', [1, 2, 4]))
self.assertTrue(s.set('length_of_ignition', [1, 2, 4]))
self.assertTrue(s.set('area', [1, 2, 4]))
self.assertTrue(s.set('ecoregion', ['western', 'southern', 'boreal']))
self.assertTrue(s.set('fm_1000hr', [1, 2, 4]))
self.assertTrue(s.set('fm_duff', [1, 2, 4]))
self.assertTrue(s.set('fm_litter', [1, 2, 4]))
self.assertTrue(s.set('can_con_pct', [1, 2, 4]))
self.assertTrue(s.set('shrub_black_pct', [1, 2, 4]))
self.assertTrue(s.set('season', ['spring','fall','winter']))
s.fm_type = 'MEAS-Th'
s.units = 'kg_ha'
result = s.display_settings()
def testAddSettings(self):
''' Add settings and check:
- need to specify burn_type first
- individual values will be turned into a sequence if necessary
- the object knows when it has all required settings
'''
s = ConsumeInputSettings()
items = [1, 2, 4]
self.assertFalse(s.set('fuelbeds', items)) # need to specify burn_type before adding settings
s.burn_type = 'activity'
self.assertTrue(s.set('fuelbeds', items))
fb = s.get('fuelbeds')
i = 0
while i < len(fb):
self.assertEqual(fb[i], str(items[i]))
i += 1
self.assertTrue(s.set('slope', [1, 2, 4]))
self.assertTrue(s.set('windspeed', [1, 2, 4]))
self.assertTrue(s.set('days_since_rain', [1, 2, 4]))
self.assertTrue(s.set('fm_10hr', 2)) ### - manufacture a sequence if necessary
self.assertTrue(s.set('length_of_ignition', [1, 2, 4]))
self.assertTrue(s.set('area', [1, 2, 4]))
self.assertTrue(s.set('ecoregion', ['western', 'southern', 'boreal']))
self.assertTrue(s.set('fm_1000hr', [1, 2, 4]))
self.assertTrue(s.set('fm_duff', [1, 2, 4]))
self.assertTrue(s.set('fm_litter', [1,2,4]))
self.assertTrue(s.set('can_con_pct', 4)) ### - manufacture a sequence if necessary
self.assertTrue(s.set('shrub_black_pct', [1, 2, 4]))
self.assertTrue(s.set('pile_black_pct', [1, 2, 4]))
self.assertTrue(s.set('season', ['spring','fall','winter']))
s.fm_type = 'MEAS-Th'
self.assertFalse(s.settings_are_complete())
s.units = 'kg_ha'
self.assertTrue(s.settings_are_complete())
def test_load_activity(self):
''' Write a temp file, load the temp file and verify that the settings are equal to the
values written. Close tempfile
'''
data = TestInputSettings.act_data
s1 = set(data.columns)
s2 = set(ConsumeInputSettings.AllSNames)
s2.add("units")
s2.add("fm_type")
self.assertEqual(s1, s2)
infile = self.write_file(data)
s = ConsumeInputSettings()
s.burn_type = 'activity'
self.assertTrue(s.load_from_file(infile[1]))
self.assertEqual(s.burn_type, 'activity')
self.assertEqual(s.units, 'kg_ha')
self.assertEqual(s.fm_type, 'NFDRS-Th')
self.assertEqual(list(s.get('area')), [float(x) for x in list(data.area)])
self.assertEqual(list(s.get('can_con_pct')), [float(x) for x in list(data.can_con_pct)])
self.assertEqual(list(s.get('ecoregion')), list(data.ecoregion))
self.assertEqual(list(s.get('fm_1000hr')), [float(x) for x in list(data.fm_1000hr)])
self.assertEqual(list(s.get('fm_duff')), [float(x) for x in list(data.fm_duff)])
self.assertEqual(list(s.get('fm_litter')), [float(x) for x in list(data.fm_litter)])
self.assertEqual(list(s.get('fuelbeds')), [str(x) for x in list(data.fuelbeds)])
self.assertEqual(list(s.get('shrub_black_pct')), [float(x) for x in list(data.shrub_black_pct)])
self.assertEqual(list(s.get('season')), [x for x in list(data.season)])
self.assertEqual(list(s.get('slope')), [float(x) for x in list(data.slope)])
self.assertEqual(list(s.get('windspeed')), [float(x) for x in list(data.windspeed)])
self.assertEqual(list(s.get('days_since_rain')), [float(x) for x in list(data.days_since_rain)])
self.assertEqual(list(s.get('fm_10hr')), [float(x) for x in list(data.fm_10hr)])
self.assertEqual(list(s.get('length_of_ignition')), [float(x) for x in list(data.length_of_ignition)])
os.unlink(infile[1])
def test_load_natural_from_dict(self):
''' Package into dict, and verify that the settings
Bluesky uses this.
'''
def as_ints(in_list):
return [int(i) for i in in_list]
data = TestInputSettings.nat_data
input_dict = {
'fuelbeds' : data.fuelbeds.values,
'area' : as_ints(data.area.values),
'units': data.units.loc[0],
'ecoregion' : data.ecoregion.values,
'fm_1000hr' : as_ints(data.fm_1000hr.values),
'fm_duff' : as_ints(data.fm_duff.values),
'fm_litter' : as_ints(data.fm_litter.values),
'can_con_pct' : as_ints(data.can_con_pct.values),
'shrub_black_pct' : as_ints(data.shrub_black_pct.values),
'pile_black_pct' : as_ints(data.pile_black_pct.values),
'season' : data.season.values
}
s = ConsumeInputSettings()
s.burn_type = 'natural'
self.assertTrue(s.load_from_dict(input_dict))
self.assertEqual(s.burn_type, 'natural')
self.assertEqual(s.units, 'kg_ha')
self.assertEqual(list(s.get('area')), [float(x) for x in list(data.area)])
self.assertEqual(list(s.get('can_con_pct')), [float(x) for x in list(data.can_con_pct)])
self.assertEqual(list(s.get('ecoregion')), list(data.ecoregion))
self.assertEqual(list(s.get('fm_1000hr')), [float(x) for x in list(data.fm_1000hr)])
self.assertEqual(list(s.get('fm_duff')), [float(x) for x in list(data.fm_duff)])
self.assertEqual(list(s.get('fm_litter')), [float(x) for x in list(data.fm_litter)])
self.assertEqual(list(s.get('fuelbeds')), [str(x) for x in list(data.fuelbeds)])
self.assertEqual(list(s.get('shrub_black_pct')), [float(x) for x in list(data.shrub_black_pct)])
self.assertEqual(list(s.get('season')), [x for x in list(data.season)])
def test_load_natural(self):
''' Write a temp file, load the temp file and verify that the settings are equal to the
values written. Close tempfile
'''
data = TestInputSettings.nat_data
s1 = set(data.columns)
s2 = set(ConsumeInputSettings.NaturalSNames)
s2.add("units")
self.assertEqual(s1, s2)
infile = self.write_file(data)
s = ConsumeInputSettings()
s.burn_type = 'natural'
self.assertTrue(s.load_from_file(infile[1]))
self.assertEqual(s.burn_type, 'natural')
self.assertEqual(s.units, 'kg_ha')
self.assertEqual(list(s.get('area')), [float(x) for x in list(data.area)])
self.assertEqual(list(s.get('can_con_pct')), [float(x) for x in list(data.can_con_pct)])
self.assertEqual(list(s.get('ecoregion')), list(data.ecoregion))
self.assertEqual(list(s.get('fm_1000hr')), [float(x) for x in list(data.fm_1000hr)])
self.assertEqual(list(s.get('fm_duff')), [float(x) for x in list(data.fm_duff)])
self.assertEqual(list(s.get('fm_litter')), [float(x) for x in list(data.fm_litter)])
self.assertEqual(list(s.get('fuelbeds')), [str(x) for x in list(data.fuelbeds)])
self.assertEqual(list(s.get('shrub_black_pct')), [float(x) for x in list(data.shrub_black_pct)])
self.assertEqual(list(s.get('season')), [x for x in list(data.season)])
os.unlink(infile[1])